Update app.py

app.py

@@ -8,41 +8,28 @@ class UnifiedSystem:
         self.sql_generator = SQLGenerator()
         self.intent_classifier = IntentClassifier()
         self.rag_system = RAGSystem()
-        self.base_url = "https://agkd0n-fa.myshopify.com/products/"  # Shopify base URL for product links
-
+        
     def process_query(self, query):
         intent, confidence = self.intent_classifier.classify(query)
-
+        
         if intent == "database_query":
-            # Generate SQL query and fetch Shopify data for database queries
             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']}, "
-                    f"Link: {self.base_url}{p['handle']}"
+                    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}"
             else:
                 return "No results found or error fetching data from Shopify."
-
+                
         elif intent == "product_description":
-            # Retrieve product description and relevant handles from RAGSystem
             rag_response = self.rag_system.process_query(query)
-            
-            # Assuming `rag_response` now includes a list of dictionaries with 'description' and 'handle' keys
-            if rag_response and 'products' in rag_response:
-                results = "\n".join([
-                    f"Description: {p['description']}, Link: {self.base_url}{p['handle']}"
-                    for p in rag_response['products']
-                ])
-                return f"Intent: Product Description (Confidence: {confidence:.2f})\n\n" \
-                       f"Response:\n{results}"
-            else:
-                return "No matching products found for the description query."
+            return f"Intent: Product Description (Confidence: {confidence:.2f})\n\n" \
+                   f"Response: {rag_response}"
         
         return "Intent not recognized."
 
@@ -52,8 +39,7 @@ def create_interface():
     examples = [
         "Show me shirt less than 5 USD",
         "Show me shirt with red color",
-        "Show me T-shirt with M size",
-        "Describe the features of the blue jeans"
+        "Show me T-shirt with M size"
     ]
     
     iface = gr.Interface(