Update app.py

app.py

@@ -2,18 +2,14 @@ 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":
@@ -36,9 +32,6 @@ 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()
@@ -51,17 +44,12 @@ 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.",
-        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"
+        description="Enter a natural language query to search products or get descriptions."
     )
     
+    # Optionally, add a button here (though not necessary, as the textbox will have submit functionality)
+    iface.add_component(gr.Button(label="Submit"))
+    
     return iface
 
 if __name__ == "__main__":