Update app.py

app.py

@@ -1,49 +1,55 @@
-import google.generativeai as palm
-import os
 import gradio as gr
-import random
+import os
 import time
+from google.generativeai import palm
+
 palm.configure(api_key=os.environ.get("palm_key"))
 
 defaults = {
-  'model': 'models/chat-bison-001',
-  'temperature': 0.25,
-  'candidate_count': 1,
-  'top_k': 40,
-  'top_p': 0.95,
+    'model': 'models/chat-bison-001',
+    'temperature': 0.25,
+    'candidate_count': 1,
+    'top_k': 40,
+    'top_p': 0.95,
 }
-context = "Your a computer failure assitant"
+
+context = "You're a computer failure assistant"
+
 examples = [
-  [
-    "Hey my computer is broken",
-    "Hey what is the issue with your computer?"
-  ]
+    [
+        "Hey my computer is broken",
+        "Hey, what is the issue with your computer?"
+    ]
 ]
-messages = []
-messages.append("NEXT REQUEST")
+
+messages = ["NEXT REQUEST"]
+
 response = palm.chat(
-  **defaults,
-  context=context,
-  examples=examples,
-  messages=messages
+    **defaults,
+    context=context,
+    examples=examples,
+    messages=messages
 )
 
 
-with gr.Blocks() as demo:
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox()
-    clear = gr.ClearButton([msg, chatbot])
+def respond(msg):
+    messages.append(msg)
+    bot_message = palm.chat(
+        **defaults,
+        context=context,
+        examples=examples,
+        messages=messages
+    )
+    time.sleep(2)
+    return bot_message['message'], messages
 
-    def respond(response, messages):
-        bot_message = palm.chat(
-          **defaults,
-          context=context,
-          examples=examples,
-          messages=messages
-        )
-        time.sleep(2)
-        return "", messages
 
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
+iface = gr.Interface(
+    fn=respond,
+    inputs="text",
+    outputs="text",
+    title="Chatbot",
+    description="Interact with a chatbot."
+)
 
-demo.launch()
+iface.launch()