Create app.py

app.py

@@ -0,0 +1,54 @@
+from models.intent_classifier import IntentClassifier
+from models.rag_system import RAGSystem
+from models.sql_generator import SQLGenerator
+from interface.gradio_interface import GradioInterface
+from utils.logger import setup_logger
+
+logger = setup_logger(__name__)
+
+class UnifiedSystem:
+    def __init__(self):
+        try:
+            self.sql_generator = SQLGenerator()
+            self.intent_classifier = IntentClassifier()
+            self.rag_system = RAGSystem()
+        except Exception as e:
+            logger.error(f"Failed to initialize UnifiedSystem: {str(e)}")
+            raise
+
+    def process_query(self, query):
+        try:
+            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")
+                
+                if products and 'products' in products:
+                    results = "\n".join([
+                        f"Title: {p['title']}, Vendor: {p['vendor']}"
+                        for p in products['products']
+                    ])
+                    return (f"Intent: Database Query (Confidence: {confidence:.2f})\n\n"
+                           f"SQL Query: {sql_query}\n\nResults:\n{results}")
+                return "No results found or error fetching data from Shopify."
+            
+            elif intent == "product_description":
+                rag_response = self.rag_system.process_query(query)
+                return (f"Intent: Product Description (Confidence: {confidence:.2f})\n\n"
+                       f"Response: {rag_response}")
+            
+            return "Intent not recognized."
+        except Exception as e:
+            logger.error(f"Query processing error: {str(e)}")
+            return f"An error occurred while processing your query: {str(e)}"
+
+if __name__ == "__main__":
+    try:
+        system = UnifiedSystem()
+        interface = GradioInterface(system)
+        iface = interface.create_interface()
+        iface.launch()
+    except Exception as e:
+        logger.error(f"Application startup failed: {str(e)}")
+