Update Chatbot.py

Chatbot.py

@@ -138,28 +138,35 @@ class ChatInterface:
     def get_chat_history(self, user_id):
         return MongoDBChatMessageHistory(
             session_id=user_id,
-            connection_string= os.getenv("Mongo_URI"),
+            connection_string=os.getenv("Mongo_URI"),
             database_name="Chatbot",
             collection_name="chat_histories",
         )
-    def chat(self, message, history, user_id):
-        if not user_id:
-            return "Please enter a user ID First.", history,
-        
-        config = {"configurable": {"thread_id": user_id}}
-        chat_history = self.get_chat_history(user_id)
 
-        response = self.model.invoke(message, config, chat_history)
-        return response, history + [(message, response)]
-    
+    def chat(self, message, history, user_id, language="English"):
+        try:
+            if not user_id:
+                return "", history, "Please enter a user ID first."
+            
+            config = {"configurable": {"thread_id": user_id}}
+            chat_history = self.get_chat_history(user_id)
+            
+            response = self.model.invoke(message, config, chat_history)
+            return "", history + [(message, response)], ""
+        except Exception as e:
+            return "", history, f"Error: {str(e)}"
+
     def reset_chat(self, user_id):
-        if user_id:
+        try:
+            if not user_id:
+                return None, "Please enter a user ID first"
             chat_history = self.get_chat_history(user_id)
             chat_history.clear()
-            return f"Chat history cleared for user {user_id}"
-        return "Please enter a user ID first"
+            return None, f"Chat history cleared for user {user_id}"
+        except Exception as e:
+            return None, f"Error clearing chat: {str(e)}"
     
-    # Create the Gradio Interface
+# Create the Gradio Interface
 def create_gradio_interface():
     chat_interface = ChatInterface()
 
@@ -190,6 +197,7 @@ def create_gradio_interface():
                 Welcome to your personal AI assistant! Enter your ID and start chatting.
                 """
             )
+            
             with gr.Row():
                 with gr.Column(scale=3):
                     user_id = gr.Textbox(
@@ -208,50 +216,42 @@ def create_gradio_interface():
             chatbot = gr.Chatbot(
                 label="💬 Chat History",
                 height=500,
-                bubble_full_width= False,
-                show_label= True,
-                elem_id = "chatbot"
+                bubble_full_width=False,
+                show_label=True,
+                elem_id="chatbot"
             )
 
             with gr.Row():
-                with gr.Column(scale=4):
-                    msg = gr.Textbox(
-                        label="✍️ Message",
-                        placeholder="Type your message Here...",
-                        show_label = True,
-                        elem_id = "message"
-                    )
-                with gr.Column(scale = 1):
-                    submit = gr.Button("🚀 Send", variant="primary")
+                msg = gr.Textbox(
+                    label="✍️ Message",
+                    placeholder="Type your message here...",
+                    show_label=True,
+                    elem_id="message"
+                )
+                submit = gr.Button("🚀 Send", variant="primary")
 
-            with gr.Row():
-                clear = gr.Button("🗑️ Clear Chat", variant="secondary")
-            
-            # Status message for feedback
+            clear = gr.Button("🗑️ Clear Chat", variant="secondary")
             status_msg = gr.Markdown("")
 
-        # Event Handler
-        msg_handler = submit.click(
-            chat_interface.chat,
-            inputs=[msg, chatbot, user_id, language],
-            outputs=[msg, chatbot, status_msg],
-            api_name = "chat"
-        )
+            # Event handlers
+            submit.click(
+                fn=chat_interface.chat,
+                inputs=[msg, chatbot, user_id, language],
+                outputs=[msg, chatbot, status_msg]
+            )
 
-        msg.submit(
-            chat_interface.chat,
-            inputs=[msg, chatbot, user_id, language],
-            outputs=[msg, chatbot, status_msg],
-            api_name="chat_submit"
-        )
+            msg.submit(
+                fn=chat_interface.chat,
+                inputs=[msg, chatbot, user_id, language],
+                outputs=[msg, chatbot, status_msg]
+            )
 
-        clear.click(
-            chat_interface.reset_chat,
-            inputs = [user_id],
-            outputs=[chatbot],
-            api_name= "clear"
-        )
-    
+            clear.click(
+                fn=chat_interface.reset_chat,
+                inputs=[user_id],
+                outputs=[chatbot, status_msg]
+            )
+  
     return demo
 
 if __name__ == "__main__":