Update app.py

app.py

@@ -6,32 +6,40 @@ from langchain_google_genai.chat_models import ChatGoogleGenerativeAI
 os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "./firm-catalyst-437006-s4-407500537db5.json"
 
 # Chat function
-def chat_with_gemini(user_input, chat_history=""):
+def chat_with_gemini(user_input, chat_history=[]):
     try:
-        # Append user input to the chat history
-        chat_history += f"User: {user_input}\n"
-
-        # Get response from the model (assuming 'predict' is the correct method)
-        response = llm.predict(user_input)
+        # Append the user input to the chat history
+        chat_history.append(("User", user_input))
+        
+        # Get response from the model
+        response = llm.predict(user_input)  # Assuming 'predict' is the correct method
         
-        # Append the response to the chat history
-        chat_history += f"Bot: {response}\n"
+        # Append the bot's response to the chat history
+        chat_history.append(("Bot", response))
 
         # Return the updated chat history
         return chat_history
     except Exception as e:
-        return chat_history + f"Error: {str(e)}\n"
+        # In case of an error, return the error message in the chat
+        chat_history.append(("Bot", f"Error: {str(e)}"))
+        return chat_history
+
+# Create a Gradio interface
+with gr.Blocks() as iface:
+    gr.Markdown("# Chatbot with Gemini 1.5")
+    gr.Markdown("Ask me anything!")
+    
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(label="Type your message here...")
+    state = gr.State([])  # Store chat history
+
+    # Define the interaction between components
+    def send_message(user_input, chat_history):
+        return chat_with_gemini(user_input, chat_history)
 
-# Create Gradio interface
-iface = gr.Interface(
-    fn=chat_with_gemini,
-    inputs=["text", "text"],  # "text" input for user message and chat history
-    outputs="text",           # Single "text" output for the chat history
-    title="Chatbot with Gemini 1.5",
-    description="Ask me anything!",
-    theme="default",
-    live=True
-)
+    # Set up the interaction for when the user submits a message
+    msg.submit(send_message, [msg, state], [chatbot, state])
+    msg.submit(lambda: "", None, msg)  # Clear input box after submission
 
 # Launch the interface with debugging enabled
 iface.launch(debug=True)