Update app.py

app.py

@@ -3,7 +3,7 @@ import os
 import time
 import google.generativeai as palm
 
-palm.configure(api_key=os.environ.get("palm_key"))
+palm.configure(api_key=api_key)
 
 defaults = {
     'model': 'models/chat-bison-001',
@@ -22,41 +22,46 @@ examples = [
     ]
 ]
 
-# user_message = ['']
 history = ['']
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     chatbot = gr.Chatbot()
     msg = gr.Textbox()
-    btn = gr.Button("Submit",variant="primary")
-    # gr.ClearButton([msg, chatbot])
-    clear = gr.Button("Clear")
+    btn = gr.Button("Submit", variant="primary")
+    gr.ClearButton([msg, chatbot])
    
-
     def user(user_message, history):
-        return gr.update(value=""), history + [[user_message,None]]
-        # , interactive=False
+        return gr.update(value=""), history + [[user_message, None]]
+    
     def bot(history):
-        # Remove the second element of the last element of the list
-        # del history[-1][1]
-        bot_message = palm.chat(
-            context=context,
-            examples=examples,
-            messages=history[-1][0]
-        )
-        history[-1][1] = ""
-        for character in bot_message.last:
-            history[-1][1] += character
-            time.sleep(0.005)
-            yield history
+        try:        
+            bot_message = palm.chat(
+                context=context,
+                examples=examples,
+                messages=history[-1][0]
+            )
+            
+            history[-1][1] = ""
+            for character in bot_message.last:
+                history[-1][1] += character
+                time.sleep(0.005)
+        except Exception as e:
+            # Handle the exception here
+            print("Error occurred:", str(e))
+            # You can customize the error handling as per your requirements
+            # For example, return an error message to the user
+            
+            history[-1][1] = "Incorrect input please retry"
+        
+        return history
+
     response = msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-        bot, chatbot,  chatbot
+        bot, chatbot, chatbot
     )
     response = btn.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-        bot, chatbot,  chatbot
+        bot, chatbot, chatbot
     )
     response.then(lambda: gr.update(interactive=True), None, [msg], queue=False)
-    clear.click(lambda: None, None, chatbot, queue=False)
 
 demo.queue()
-demo.launch()
+demo.launch(debug=True)