Update app.py

app.py

@@ -2,14 +2,18 @@ import gradio as gr
 from sql_generator import SQLGenerator
 from intent_classifier import IntentClassifier
 from rag_system import RAGSystem
+import threading
 
 class UnifiedSystem:
     def __init__(self):
         self.sql_generator = SQLGenerator()
         self.intent_classifier = IntentClassifier()
         self.rag_system = RAGSystem()
-        
+        self.current_query_thread = None  # To keep track of the current query thread
+        self.stop_event = threading.Event()  # Event to stop the current process
+
     def process_query(self, query):
+        self.stop_event.clear()  # Clear the stop event
         intent, confidence = self.intent_classifier.classify(query)
         
         if intent == "database_query":
@@ -32,6 +36,9 @@ class UnifiedSystem:
                    f"Response: {rag_response}"
         
         return "Intent not recognized."
+    
+    def stop_query(self):
+        self.stop_event.set()  # Set the event to stop the current process
 
 def create_interface():
     system = UnifiedSystem()
@@ -44,11 +51,19 @@ def create_interface():
         ),
         outputs=gr.Textbox(label="Response"),
         title="Unified Query Processing System",
-        description="Enter a natural language query to search products or get descriptions."
+        description="Enter a natural language query to search products or get descriptions.",
+        live=False  # Live updates are not needed for this scenario
+    )
+
+    # Add a button to stop the current query
+    iface.add_button(
+        "Kill Query",
+        lambda: system.stop_query(),
+        elem_id="stop-button"
     )
     
     return iface
 
 if __name__ == "__main__":
     iface = create_interface()
-    iface.launch()
+    iface.launch()