Update app.py

app.py

@@ -352,36 +352,49 @@ class OpenRouterLLM:
 
 
 class UserSession:
-    def __init__(self):
+    def __init__(self, llm: OpenRouterLLM):  # Accept an instance of OpenRouterLLM
         self.current_user = None
         self.welcome_message = None
         self.conversation_history = []  # Add conversation history storage
-
+        self.llm = llm  # Store the LLM instance
+    
     def set_user(self, user_info):
         self.current_user = user_info
         self.set_welcome_message(user_info.get("Nickname", "Guest"))
         # Initialize conversation history with welcome message
         welcome = self.get_welcome_message()
-        #initial_message = (" "
-        #)
         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, Nickname):
+        """Set a dynamic welcome message using the OpenRouterLLM."""
+        # Define a prompt for the LLM to generate a welcome message
+        prompt = (
+            f"Create a very brief welcome message for {Nickname} that fits in 3 lines. "
+            f"The message should: "
+            f"1. Welcome {Nickname} warmly and professionally. "
+            f"2. Emphasize that this is a safe and trusted space. "
+            f"3. Highlight specialized support for gender-based violence (GBV) and legal assistance. "
+            f"4. Use a tone that is warm, reassuring, and professional. "
+            f"5. Keep the message concise and impactful, ensuring it fits within the character limit."
+        )
+        
+        # Use the OpenRouterLLM to generate the message
+        welcome_text = "".join(self.llm.stream(prompt))  # Stream and concatenate the response
+        
+        # Format the message with HTML styling
         self.welcome_message = (
             f"<div style='font-size: 24px; font-weight: bold; color: #2E86C1;'>"
             f"Welcome {Nickname}! 👋</div>"
-            f"<div style='font-size: 20px'>"
-            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"<div style='font-size: 20px;'>"
+            f"{welcome_text}"
             f"</div>"
         )
-
+    
     def get_welcome_message(self):
         return self.welcome_message
     
@@ -401,8 +414,9 @@ class UserSession:
             formatted_history += f"{role}: {entry['content']}\n\n"
         return formatted_history
 
-# Initialize session
-user_session = UserSession()
+api_key =api  
+llm_instance = OpenRouterLLM(key=api_key)  
+user_session = UserSession(llm_instance)
 
 # Store user details and handle session
 def collect_user_info(Nickname):
@@ -581,7 +595,7 @@ def chatbot_interface():
     with gr.Blocks() as demo:
         # User registration section
         with gr.Column(visible=True, elem_id="registration_container") as registration_container:
-            gr.Markdown("### Your privacy is our concern, please provide your nickname.")
+            gr.Markdown("### Your privacy matters to us! Just share a nickname you feel comfy with to start chatting..")
 
             with gr.Row():
                 first_name = gr.Textbox(
@@ -605,7 +619,7 @@ def chatbot_interface():
             )
 
             # Footer with version info
-            gr.Markdown("Ijwi ry'Ubufasha_Chatbot v1.0.0 © 2025")
+            gr.Markdown("Ijwi ry'Ubufasha Chatbot v1.0.1 © 2025")
 
         # Handle user registration
         submit_btn.click(