Update app.py

app.py

@@ -11,8 +11,16 @@ title = '🦒Playground w/ Google PaLM v2'
 description = """Click below tab and start with your message"""
 
 def generate_text(prompt: str):
-    response = genai.generate_text(prompt=prompt)
-    return response.result
+    print("\n")
+    print(f"User Message:\n{prompt}\n")
+    try:
+        response = genai.generate_text(prompt=prompt)
+        result = response.result
+    except Exception as e:
+        result = "Something went wrong. Please try again with other prompts."
+        print(f"Exception {e} occured\n")
+    print(f"Bot Message:\n{result}\n")
+    return result
 
 chat_defaults = {
   'model': 'models/chat-bison-001',
@@ -26,14 +34,24 @@ chat_messages = []
 
 def generate_chat(prompt: str):
     context = "You are an intelligent chatbot powered by biggest technology company."
+    print("\n")
+    print(f"User Message:\n{prompt}\n")
     chat_messages.append(prompt)
-    response = genai.chat(
-      **chat_defaults,
-      context=context,
-      messages=chat_messages
-    )
-    chat_messages.append(response.last)
-    return response.last
+    
+    try:
+        response = genai.chat(
+          **chat_defaults,
+          context=context,
+          messages=chat_messages
+        )
+        result = response.last
+        chat_messages.append(result)
+    except Exception as e:
+        result = "Something went wrong. Please try again with other messages."
+        chat_messages = chat_messages[:-1]
+        print(f"Exception {e} occured\n")
+    print(f"Bot Message:\n{result}\n")
+    return result
 
 with gr.Blocks(theme='JohnSmith9982/small_and_pretty') as demo:
     
@@ -56,9 +74,7 @@ with gr.Blocks(theme='JohnSmith9982/small_and_pretty') as demo:
             return "", history + [[user_message, None]]
     
         def bot(history):
-            print(f"User Message: {history[-1][0]}\n")
             bot_message = generate_chat(history[-1][0])
-            print(f"Bot Message: {bot_message}")
             history[-1][1] = ""
             for character in bot_message:
                 history[-1][1] += character