Update app.py

app.py

@@ -1,28 +1,30 @@
 import gradio as gr
 from chromadb_semantic_search_for_dataset import search_cases, chat_with_cases
 
-# Global variable to store current case context
+# Global variables to store current context
 current_cases_context = ""
+current_user_case = ""
 
 def search_and_store_cases(query):
     """Search for cases and store context for chat"""
-    global current_cases_context
+    global current_cases_context, current_user_case
     if not query.strip():
         return "कृपया केस खोज्नका लागि केही लेख्नुहोस्।", ""
     
-    formatted_results, context = search_cases(query)
+    formatted_results, context, user_case = search_cases(query)
     current_cases_context = context
+    current_user_case = user_case
     return formatted_results, ""
 
 def chat_function(message, history):
     """Main chat function that uses current case context"""
-    global current_cases_context
+    global current_cases_context, current_user_case
     
     if not message.strip():
         return history, ""
     
-    # Generate response using current case context
-    bot_response = chat_with_cases(message, current_cases_context)
+    # Generate response using current case context and user's case
+    bot_response = chat_with_cases(message, current_cases_context, current_user_case)
     
     # Add to chat history
     history.append([message, bot_response])
@@ -30,8 +32,9 @@ def chat_function(message, history):
 
 def clear_context():
     """Clear the stored case context"""
-    global current_cases_context
+    global current_cases_context, current_user_case
     current_cases_context = ""
+    current_user_case = ""
     return "केस कन्टेक्स्ट सफा गरियो। अब तपाईं सामान्य च्याट गर्न सक्नुहुन्छ।", [], ""
 
 # Custom CSS
@@ -153,12 +156,14 @@ with gr.Blocks(css=css, title="नेपाली कानूनी सहा
                 
                 gr.Examples(
                     examples=[
-                        ["यी केसहरूमा के फैसला भएको थियो?"],
-                        ["मेरो केसमा के गर्न सकिन्छ?"],
+                        ["मेरो केसमा के फैसला हुन सक्छ?"],
+                        ["यी केसहरूसँग मेरो मुद्दाको समानता के छ?"],
+                        ["मेरो केसको भिन्नता के छ?"],
                         ["यस्ता मुद्दामा कति समय लाग्छ?"],
-                        ["अदालती फी कति लाग्छ?"],
-                        ["वकिल किन आवश्यक छ?"],
-                        ["प्रमाण कसरी जम्माउने?"]
+                        ["के प्रमाणहरू चाहिन्छ?"],
+                        ["मेरो केसमा जित्ने सम्भावना कति छ?"],
+                        ["यी फैसलाहरूबाट के सिक्न सकिन्छ?"],
+                        ["अदालती प्रक्रिया कस्तो हुन्छ?"]
                     ],
                     inputs=msg_input,
                     label="च्याट उदाहरणहरू (Chat Examples)"
@@ -186,7 +191,7 @@ with gr.Blocks(css=css, title="नेपाली कानूनी सहा
         inputs=[search_input],
         outputs=[search_results, context_status]
     ).then(
-        fn=lambda: "केस कन्टेक्स्ट लोड भयो। अब च्याट गर्न सक्नुहुन्छ।",
+        fn=lambda: "तपाईंको केस र समान केसहरूको कन्टेक्स्ट लोड भयो। अब तुलनाका लागि च्याट गर्न सक्नुहुन्छ।",
         outputs=[context_status]
     )