Update app.py

app.py

@@ -7,20 +7,6 @@ import google.generativeai as genai
 genai.configure(api_key=os.getenv('PALM_API_KEY'))
 
 # Gradio
-title = '🦒Playground w/ Google PaLM v2'
-description = """Click below tab and start with your message in English"""
-
-def generate_text(prompt: str):
-    print("Generating Text...")
-    print(f"User Message:\n{prompt}\n")
-    try:
-        response = genai.generate_text(prompt=prompt)
-        result = response.result
-    except Exception as e:
-        result = "Apologies but something went wrong. Please try again later."
-        print(f"Exception {e} occured\n")
-    print(f"Bot Message:\n{result}\n")
-    return result
 
 chat_defaults = {
   'model': 'models/chat-bison-001',
@@ -32,69 +18,43 @@ chat_defaults = {
 
 chat_history = []
 
-def generate_chat(prompt: str, chat_messages=chat_history):
-    print(chat_messages)
+def generate_chat(message: str, history=chat_history):
+    print(history)
     context = "You are an intelligent chatbot powered by biggest technology company."
     print("Generating Chat Message...")
-    print(f"User Message:\n{prompt}\n")
-    chat_messages.append(prompt)
+    print(f"User Message:\n{message}\n")
+    history.append(message)
+    
     try:
         response = genai.chat(
           **chat_defaults,
           context=context,
-          messages=chat_messages
+          messages=history
         )
         result = response.last
         if result is None:
             result = "Apologies but something went wrong. Please try again later."
-            chat_messages = chat_messages[:-1]
+            history = history[:-1]
         else:
-            chat_messages.append(result)
+            history.append(result)
     except Exception as e:
         result = "Apologies but something went wrong. Please try again later."
-        chat_messages = chat_messages[:-1]
-        print(f"Exception {e} occured\n")
-    chat_history = chat_messages
+        history = history[:-1]
+        print(f"Exception {e}\n")
+        
+    chat_history = history
     print(f"Bot Message:\n{result}\n")
-    return result
-
-with gr.Blocks(theme='JohnSmith9982/small_and_pretty') as demo:
-    gr.Markdown(
-    f"""
-    # {title}
-    ### {description}
-    """)
 
-    with gr.Tab('Just Text'):
-        app = gr.Interface(fn=generate_text, inputs=["text"], outputs=["text"],
-                           examples=[["What is Large Language Model?"], ["Write me a story about Pokemon."]]
-        )
-
-    with gr.Tab('Just Chat'):
-        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__
-    """)
+    for i in range(result):
+        time.sleep(0.01)
+        yield result[: i+1]
+
+app = gr.ChatInterface(
+        generate_chat,
+        additional_inputs=additional_inputs,
+        examples=[["What is the secret to life?"], ["Write me a recipe for pancakes."]]
+    )
     
-demo.queue()
-demo.launch()
+app.queue()
+app.launch()