Update app.py

app.py

@@ -411,7 +411,7 @@ class GBVSupportChatbot:
         # Combine all data
         all_data = []
         all_data.extend(context_data)
-        #all_data.extend([item for item in text_chunks if item not in all_data])
+        all_data.extend([item for item in text_chunks if item not in all_data])
         
         # Add data to vector store
         self.vectorstore.add_texts(all_data)
@@ -463,16 +463,16 @@ class GBVSupportChatbot:
         # Generate welcome message
         welcome_message = self.user_session.get_welcome_message()
         
-        # Add initial message to start the conversation
+        # For the standard Chatbot component, the history format is different
         chat_history = [(None, welcome_message)]
         
         # Return welcome message and update UI
         return welcome_message, gr.update(visible=True), gr.update(visible=False), chat_history
     
-    def rag_memory_stream(self, message: str, history, user_lang: str = "kin_Latn", system_lang: str = "eng_Latn"):
+    def rag_memory_stream(self, message: str, history):
         """Process user message, translate, and generate response."""
-        # Translate user message to English
-        english_message = self.translator.translate_text(message, user_lang, system_lang)
+        # Translate user message to English (from Kinyarwanda by default)
+        english_message = self.translator.translate_text(message, "kin_Latn", "eng_Latn")
         
         # Add translated message to history
         self.user_session.add_to_history("user", english_message)
@@ -484,8 +484,8 @@ class GBVSupportChatbot:
         for new_text in rag_chain({"question": english_message}):
             full_response += new_text
         
-        # Translate response back to user language
-        translated_response = self.translator.translate_text(full_response, system_lang, user_lang)
+        # Translate response back to user language (Kinyarwanda by default)
+        translated_response = self.translator.translate_text(full_response, "eng_Latn", "kin_Latn")
         
         # Add response to history
         self.user_session.add_to_history("assistant", full_response)
@@ -514,10 +514,43 @@ class GBVSupportChatbot:
             
             # Chatbot section (initially hidden)
             with gr.Column(visible=False, elem_id="chatbot_container") as chatbot_container:
-                chat_interface = gr.ChatInterface(
-                    fn=self.rag_memory_stream,
-                    title="Chat with GBVR",
-                    fill_height=True
+                # Replace ChatInterface with standard Chatbot
+                chatbot = gr.Chatbot(
+                    label="Chat with GBVR",
+                    height=500,
+                    show_label=True,
+                    elem_id="chat_interface"
+                )
+                
+                with gr.Row():
+                    msg = gr.Textbox(
+                        placeholder="Type your message here...",
+                        label="Your message",
+                        show_label=False,
+                        container=False,
+                        scale=7
+                    )
+                    send_btn = gr.Button("Send", variant="primary", scale=1)
+                
+                # Set up the chat behavior
+                def chat_response(message, history):
+                    # Clear input
+                    yield history, ""
+                    # Process message and generate response
+                    for response in self.rag_memory_stream(message, history):
+                        history.append((message, response))
+                        yield history, ""
+                
+                # Configure event handlers
+                msg.submit(
+                    chat_response,
+                    inputs=[msg, chatbot],
+                    outputs=[chatbot, msg]
+                )
+                send_btn.click(
+                    chat_response,
+                    inputs=[msg, chatbot],
+                    outputs=[chatbot, msg]
                 )
                 
                 # Footer with version info
@@ -527,7 +560,7 @@ class GBVSupportChatbot:
             submit_btn.click(
                 self.collect_user_info,
                 inputs=[first_name],
-                outputs=[response_message, chatbot_container, registration_container, chat_interface.chatbot]
+                outputs=[response_message, chatbot_container, registration_container, chatbot]
             )
         
         # Add CSS styles