Update app.py

app.py

@@ -1,35 +1,21 @@
-# main.py
 import gradio as gr
 from sql_generator import SQLGenerator
 from intent_classifier import IntentClassifier
 from rag_system import RAGSystem
-import time
 
 class UnifiedSystem:
     def __init__(self):
         self.sql_generator = SQLGenerator()
         self.intent_classifier = IntentClassifier()
         self.rag_system = RAGSystem()
-        self.stop_processing = False  # Flag to stop processing
-        
-    def stop_query(self):
-        self.stop_processing = True  # Set the flag to stop processing
         
     def process_query(self, query):
-        self.stop_processing = False  # Reset the stop flag
         intent, confidence = self.intent_classifier.classify(query)
         
         if intent == "database_query":
             sql_query = self.sql_generator.generate_query(query)
             products = self.sql_generator.fetch_shopify_data("products")
             
-            # Simulating long processing time
-            time.sleep(2)  # Replace this with actual processing logic
-
-            # Check if the process has been stopped
-            if self.stop_processing:
-                return "Query processing was stopped by the user."
-
             if products and 'products' in products:
                 results = "\n".join([
                     f"Title: {p['title']}, Vendor: {p['vendor']}" 
@@ -42,11 +28,6 @@ class UnifiedSystem:
                 
         elif intent == "product_description":
             rag_response = self.rag_system.process_query(query)
-            
-            # Check if the process has been stopped
-            if self.stop_processing:
-                return "Query processing was stopped by the user."
-
             return f"Intent: Product Description (Confidence: {confidence:.2f})\n\n" \
                    f"Response: {rag_response}"
         
@@ -57,23 +38,17 @@ def create_interface():
     
     iface = gr.Interface(
         fn=system.process_query,
-        inputs=[
-            gr.Textbox(
-                label="Enter your query",
-                placeholder="e.g., 'Show me all T-shirts' or 'Describe the product features'"
-            ),
-            gr.Button("Stop Processing")  # Add a button to stop the process
-        ],
+        inputs=gr.Textbox(
+            label="Enter your query",
+            placeholder="e.g., 'Show me all T-shirts' or 'Describe the product features'"
+        ),
         outputs=gr.Textbox(label="Response"),
         title="Unified Query Processing System",
         description="Enter a natural language query to search products or get descriptions."
     )
     
-    # Bind the stop_query method to the button
-    iface.add_event("Stop Processing", system.stop_query)
-
     return iface
 
 if __name__ == "__main__":
     iface = create_interface()
-    iface.launch(share=True)  # Set share=True to create a public link
+    iface.launch(