Update app.py

app.py

@@ -356,75 +356,88 @@ class UserSession:
     def __init__(self):
         self.current_user = None
         self.welcome_message = None
-    
+        self.conversation_history = []  # Add conversation history storage
+
     def set_user(self, user_info):
         self.current_user = user_info
-        self.set_welcome_message(user_info.get("first_name", "Guest"))
-    
+        self.set_welcome_message(user_info.get("Nickname", "Guest"))
+        # Initialize conversation history with welcome message
+        welcome = self.get_welcome_message()
+        initial_message = (
+            "<div style='font-size: 20px; font-weight: normal; color: #FFFFFF;'>"
+            f"If you are going through something, please know you are not alone, I am here for you, no matter what.🤗"
+            "</div>"
+        )
+        self.conversation_history = [
+            {"role": "assistant", "content": welcome},
+            {"role": "assistant", "content": initial_message}
+        ]
+
     def get_user(self):
         return self.current_user
-    
-    def set_welcome_message(self, first_name):
+
+    def set_welcome_message(self, Nickname):
         self.welcome_message = (
-            f"<div style='font-size: 20px; font-weight: bold; color: #2E86C1;'>"
-            f"Welcome {first_name}! 👋</div>"
-            f"<div style='font-size: 18px; color: #34495E;'>"
+            f"<div style='font-size: 24px; font-weight: bold; color: #2E86C1;'>"
+            f"Welcome {Nickname}! 👋</div>"
+            f"<div style='font-size: 20px; color: #FFFFFF;'>"
             f"We appreciate you reaching out to us. You are in a safe and trusted space designed to support you. "
-            f"Here, you can find guidance on gender-based violence (GBV) and legal assistance.<br><br>" 
+            f"Here, you can find guidance on gender-based violence (GBV) and legal assistance.<br><br>"
             f"</div>"
         )
 
-            
-           # f"You don’t have to go through this alone—we are here to listen, support, and help you find the right solutions. "
-           # f"You deserve to be heard and helped, and we are committed to standing by your side."
-            #f"</div>"
-    
     def get_welcome_message(self):
         return self.welcome_message
+    
+    def add_to_history(self, role, message):
+        self.conversation_history.append({"role": role, "content": message})
+    
+    def get_conversation_history(self):
+        return self.conversation_history
+    
+    def get_formatted_history(self):
+        formatted_history = ""
+        for entry in self.conversation_history:
+            role = "User" if entry["role"] == "user" else "Assistant"
+            formatted_history += f"{role}: {entry['content']}\n\n"
+        return formatted_history
 
 # Initialize session
 user_session = UserSession()
 
 # Store user details and handle session
-def collect_user_info(first_name, last_name, phone):
-    if not first_name or not last_name or not phone:
-        return "All fields are required to proceed.", gr.update(visible=False), gr.update(visible=True), []
-    
-    # Validate phone number (basic validation)
-    if not phone.replace("+", "").replace("-", "").replace(" ", "").isdigit():
-        return "Please enter a valid phone number.", gr.update(visible=False), gr.update(visible=True), []
-    
+def collect_user_info(Nickname):
+    if not Nickname:
+        return "Nickname is required to proceed.", gr.update(visible=False), gr.update(visible=True), []
+
     # Store user info for chat session
     user_info = {
-        "first_name": first_name,
-        "last_name": last_name,
-        "phone": phone,
+        "Nickname": Nickname,
         "timestamp": time.strftime("%Y-%m-%d %H:%M:%S")
     }
-    
+
     # Set user in session
     user_session.set_user(user_info)
-    
+
     # Generate welcome message
     welcome_message = user_session.get_welcome_message()
-    
+
     # Add initial message to start the conversation
     chat_history = add_initial_message([(None, welcome_message)])
-    
+
     # Return welcome message and update UI
     return welcome_message, gr.update(visible=True), gr.update(visible=False), chat_history
 
 # Add initial message to start the conversation
 def add_initial_message(chatbot):
     initial_message = (
-        "<div style='font-size: 16px; font-weight: normal; color: #16A085;'>"
-        f"I just want to check in and see how you are doing." 
+        "<div style='font-size: 20px; font-weight: normal; color: #FFFFFF;'>"
         f"If you are going through something, please know you are not alone, I am here for you, no matter what.🤗"
         "</div>"
     )
     return chatbot + [(None, initial_message)]
 
-# Create RAG chain with user context
+# Create RAG chain with user context and conversation history
 def create_rag_chain(retriever, template, api_key):
     llm = OpenRouterLLM(api_key)
     rag_prompt = PromptTemplate.from_template(template)
@@ -433,16 +446,20 @@ def create_rag_chain(retriever, template, api_key):
         # Get context using the retriever's invoke method
         context = retriever.invoke(input_dict["question"])
         context_str = "\n".join([doc.page_content for doc in context])
-        
+
         # Get user info from the session
         user_info = user_session.get_user() or {}
-        first_name = user_info.get("first_name", "User")
+        first_name = user_info.get("Nickname", "User")
         
-        # Format prompt with user context
+        # Get conversation history
+        conversation_history = user_session.get_formatted_history()
+
+        # Format prompt with user context and conversation history
         prompt = rag_prompt.format(
             context=context_str,
             question=input_dict["question"],
-            first_name=first_name
+            first_name=first_name,
+            conversation_history=conversation_history
         )
 
         # Stream response
@@ -451,34 +468,50 @@ def create_rag_chain(retriever, template, api_key):
     return stream_func
 
 def rag_memory_stream(message, history):
+    # Add user message to history
+    user_session.add_to_history("user", message)
+    
     # Initialize with empty response
     partial_text = ""
-    
-    # Get user context
-    user_info = user_session.get_user()
-    
+    full_response = ""
+
     # Use the rag_chain with the question
     for new_text in rag_chain({"question": message}):
         partial_text += new_text
+        full_response = partial_text
         yield partial_text
+    
+    # After generating the complete response, add it to history
+    user_session.add_to_history("assistant", full_response)
 
 # Gradio Interface Setup with improved UX
 def chatbot_interface():
     # Get API key (in a real application, handle this more securely)
     api_key = api  # This should be properly defined or imported elsewhere
 
+    # Update the template to include conversation history
+    global template
+    template = """
+    You are a compassionate and supportive AI assistant specializing in helping individuals affected by Gender-Based Violence (GBV).
+    
+    Previous conversation:
+    {conversation_history}
+    
+    Context information:
+    {context}
+    
+    User {first_name} asks: {question}
+    
+    Respond with empathy and provide support and resources based on the conversation and context. Always maintain a warm, supportive tone.
+    """
+
     # Create the RAG chain with user context
     global rag_chain
     rag_chain = create_rag_chain(retriever, template, api_key)
-    
-    # Create theme
-    theme = gr.themes.Soft(
-        primary_hue="indigo",
-        secondary_hue="blue",
-    )
-    
-    with gr.Blocks(theme=theme, css="""
-        .welcome-container { 
+
+
+    with gr.Blocks(css="""
+        .welcome-container {
             text-align: center;
             margin-bottom: 20px;
             padding: 20px;
@@ -488,74 +521,45 @@ def chatbot_interface():
         .feedback-btn { margin-top: 10px; }
         footer { margin-top: 30px; text-align: center; }
     """) as demo:
-        # Welcome banner
-        gr.Markdown("# 🤖 Ijwi ry'Ubufasha - Your AI Assistant", elem_classes=["welcome-container"])
-        
         # User registration section
         registration_container = gr.Column(visible=True)
         with registration_container:
-            gr.Markdown("### Please provide your details to start chatting")
-            
+            gr.Markdown("### To prioritize your privacy, please provide your nickname. ")
+
             with gr.Row():
                 first_name = gr.Textbox(
-                    label="First Name", 
-                    placeholder="Enter your first name",
+                    label="Nickname",
+                    placeholder="Enter your Nickname",
                     scale=1
                 )
-                last_name = gr.Textbox(
-                    label="Last Name", 
-                    placeholder="Enter your last name",
-                    scale=1
-                )
-            
-            phone = gr.Textbox(
-                label="Phone Number", 
-                placeholder="Enter your phone number (e.g., +250...)",
-            )
-            
+               
             with gr.Row():
                 submit_btn = gr.Button("Start Chatting", variant="primary", scale=2)
-            
+
             response_message = gr.Markdown(elem_id="welcome-message")
-        
+
         # Chatbot section (initially hidden)
         chatbot_container = gr.Column(visible=False)
         with chatbot_container:
             chat_interface = gr.ChatInterface(
                 fn=rag_memory_stream,
-                title="🤖 Help Chatbot",
+                title="Chat with Me",
                 fill_height=True,
-                theme=theme
             )
-            
-            # Feedback buttons
-            with gr.Row():
-                feedback_label = gr.Markdown("### Was this conversation helpful?")
-            
-            with gr.Row():
-                thumbs_up = gr.Button("👍 Yes, it was helpful", elem_classes=["feedback-btn"])
-                thumbs_down = gr.Button("👎 No, it needs improvement", elem_classes=["feedback-btn"])
-            
+
             # Footer with version info
             gr.Markdown("Ijwi ry'Ubufasha v1.0.0 © 2025", elem_id="footer")
-        
+
         # Handle user registration
         submit_btn.click(
-            collect_user_info, 
-            inputs=[first_name, last_name, phone], 
+            collect_user_info,
+            inputs=[first_name],
             outputs=[response_message, chatbot_container, registration_container, chat_interface.chatbot]
         )
-        
-        # Handle feedback (placeholder functionality)
-        def record_feedback(feedback_type):
-            # Here you could log feedback to a file or database
-            feedback_message = f"Thank you for your feedback! We'll use it to improve our service."
-            return feedback_message
-        
-        thumbs_up.click(lambda: record_feedback("positive"), outputs=feedback_label)
-        thumbs_down.click(lambda: record_feedback("negative"), outputs=feedback_label)
-    
+      
     return demo
 
+# Launch the interface
 if __name__ == "__main__":
+    # Launch the interface
     chatbot_interface().launch(share=True, inbrowser=True)