Update app.py

app.py

@@ -8,7 +8,6 @@ import google.generativeai as genai
 genai.configure(api_key=os.getenv('PALM_API_KEY'))
 
 # Gradio
-
 chat_defaults = {
   'model': 'models/chat-bison-001',
   'temperature': 0.25,
@@ -17,44 +16,65 @@ chat_defaults = {
   'top_p': 0,
 }
 
-def generate_chat(message, history):
-    message_history = list(itertools.chain(*history))
-    message_history.append(message)
-    print(message_history)
-    print(history)
+chat_history = []
+
+def generate_chat(prompt: str, chat_messages=chat_history):
+    print(chat_messages)
     context = "You are an intelligent chatbot powered by biggest technology company."
     print("Generating Chat Message...")
-    print(f"User Message:\n{message}\n")
-    
+    print(f"User Message:\n{prompt}\n")
+    chat_messages.append(prompt)
     try:
         response = genai.chat(
           **chat_defaults,
           context=context,
-          messages=message_history
+          messages=chat_messages
         )
-        print(f"\n{response}\n")
         result = response.last
         if result is None:
             result = "Apologies but something went wrong. Please try again later."
-            # chat_history = chat_history[:-1]
+            chat_messages = chat_messages[:-1]
         else:
-            # chat_history.append(result)
-            pass
+            chat_messages.append(result)
     except Exception as e:
         result = "Apologies but something went wrong. Please try again later."
-        # chat_history = chat_history[:-1]
-        print(f"Exception {e}\n")
-
+        chat_messages = chat_messages[:-1]
+        print(f"Exception {e} occured\n")
+    chat_history = chat_messages
     print(f"Bot Message:\n{result}\n")
+    return result
 
-    for i in range(len(result)):
-        time.sleep(0.01)
-        yield result[: i+1]
+with gr.Blocks(theme='HaleyCH/HaleyCH_Theme') as app:
+    gr.Markdown(
+    f"""
+    # {title}
+    ### {description}
+    """)
 
-app = gr.ChatInterface(
-        generate_chat,
-        examples=[["What is Large Language Model?"], ["Write me a story about Pokemon."]]
+    chatbot = gr.Chatbot(height=500)
+    msg = gr.Textbox()
+    clear = gr.Button("Clear")
+
+    def user(user_message, history):
+        return "", history + [[user_message, None]]
+
+    def bot(history):
+        bot_message = generate_chat(history[-1][0])
+        history[-1][1] = ""
+        for character in bot_message:
+            history[-1][1] += character
+            time.sleep(0.01)
+            yield history
+
+    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot, chatbot, chatbot
     )
+    clear.click(lambda: None, None, chatbot, queue=False)
+    
+    gr.Markdown(
+    f"""
+    Testing by __G__
+    """)
     
 app.queue()
 app.launch()