Update app.py

app.py

@@ -8,7 +8,8 @@ class UnifiedSystem:
         self.sql_generator = SQLGenerator()
         self.intent_classifier = IntentClassifier()
         self.rag_system = RAGSystem()
-        
+        self.base_url = "https://agkd0n-fa.myshopify.com/products/"
+
     def process_query(self, query):
         intent, confidence = self.intent_classifier.classify(query)
         
@@ -17,9 +18,8 @@ class UnifiedSystem:
             products = self.sql_generator.fetch_shopify_data("products")
             
             if products and 'products' in products:
-                base_url = "https://agkd0n-fa.myshopify.com/products/"
                 results = "\n".join([
-                    f"Title: {p['title']}, Vendor: {p['vendor']}, Link: {base_url}{p['handle']}" 
+                    f"Title: {p['title']}, Vendor: {p['vendor']}, URL: {self.base_url}{p['handle']}" 
                     for p in products['products']
                 ])
                 return f"Intent: Database Query (Confidence: {confidence:.2f})\n\n" \
@@ -28,32 +28,20 @@ class UnifiedSystem:
                 return "No results found or error fetching data from Shopify."
                 
         elif intent == "product_description":
-            # Assuming the RAGSystem can return relevant product information
             rag_response = self.rag_system.process_query(query)
-            description = rag_response.get("description", "No description available.")
-            # Assuming you can fetch product handles relevant to the description
-            handles = rag_response.get("handles", [])
-            base_url = "https://agkd0n-fa.myshopify.com/products/"
+            # Assume the RAG system can return product handles for links
+            product_handles = rag_response.get('product_handles', [])
+            urls = [f"{self.base_url}{handle}" for handle in product_handles]
+            response = rag_response.get('response', "No description available.")
             
-            # Generate links for each handle
-            handle_links = "\n".join([f"Link: {base_url}{handle}" for handle in handles])
-
-            return (f"Intent: Product Description (Confidence: {confidence:.2f})\n\n" 
-                    f"Response: {description}\n\n" 
-                    f"Relevant Product Links:\n{handle_links}" if handles else "No relevant product links available.")
+            return f"Intent: Product Description (Confidence: {confidence:.2f})\n\n" \
+                   f"Response: {response}\n\nProduct URLs:\n" + "\n".join(urls)
         
         return "Intent not recognized."
 
 def create_interface():
     system = UnifiedSystem()
     
-    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"
-    ]
-    
     iface = gr.Interface(
         fn=system.process_query,
         inputs=gr.Textbox(
@@ -63,7 +51,11 @@ 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.",
-        examples=examples
+        examples=[
+            ["Show me shirts less than 5 USD"],
+            ["Show me shirts with red color"],
+            ["Show me T-shirts with M size"]
+        ]
     )
     
     return iface