Update app.py

app.py

@@ -8,6 +8,7 @@ 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)
@@ -18,7 +19,8 @@ class UnifiedSystem:
             
             if products and 'products' in products:
                 results = "\n".join([
-                    f"Title: {p['title']}, Vendor: {p['vendor']}" 
+                    f"Title: {p['title']}, Vendor: {p['vendor']}, "
+                    f"Link: {self.base_url}{p['handle']}" 
                     for p in products['products']
                 ])
                 return f"Intent: Database Query (Confidence: {confidence:.2f})\n\n" \
@@ -28,8 +30,16 @@ class UnifiedSystem:
                 
         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}"
+            
+            # Assuming rag_response includes product descriptions and handles
+            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\nResults:\n{results}"
+            else:
+                return "No matching products found for the description query."
         
         return "Intent not recognized."
 
@@ -39,7 +49,8 @@ def create_interface():
     examples = [
         "Show me shirt less than 5 USD",
         "Show me shirt with red color",
-        "Show me T-shirt with M size"
+        "Show me T-shirt with M size",
+        "Describe the features of the blue jeans"
     ]
     
     iface = gr.Interface(