Update app.py

app.py

@@ -388,7 +388,7 @@ def log_query_analytics(query, product_type, response_time):
 
 # Get product images from Astra DB
 def get_product_images(product):
-    """Get product images from Astra DB"""
+    """Get product images from Astra DB, save them temporarily, and serve them"""
     global product_images
     
     if not astra_session:
@@ -405,13 +405,23 @@ def get_product_images(product):
         
         rows = astra_session.execute(query, (product,))
         
-        # Store image URLs (or IDs) for display
+        # Store image URLs for display
         image_urls = []
         for row in rows:
-            # In a real implementation, you would save the image temporarily and serve it
-            # For this demo, we're just using the image ID as an identifier
             image_id = row.id
-            image_urls.append(f"image-{image_id[:8]}")
+            image_data = row.image_data
+            
+            # Save image data to a temporary file
+            temp_dir = os.path.join(os.getcwd(), 'temp_images')
+            os.makedirs(temp_dir, exist_ok=True)
+            
+            temp_path = os.path.join(temp_dir, f"image-{image_id}.jpg")
+            with open(temp_path, 'wb') as f:
+                f.write(image_data)
+            
+            # Create a URL that can be served by your web server
+            image_url = f"/temp_images/image-{image_id}.jpg"
+            image_urls.append(image_url)
         
         # If no images found, use placeholder URLs
         if not image_urls:
@@ -823,7 +833,7 @@ def create_gradio_app():
                         <li>Ask about installation and maintenance</li>
                         <li>Get help with troubleshooting</li>
                         <li>S3 Bucket Name=agent-product-discovery</li>
-                        <li>ABB Ability™ System 800xA® 6.2.pdf, Enclosed Softstarters.pdf, Ex-Solutions.pdf, Low_power_UPS_catalogue_EN.pdf<li>
+                        <li>ABB Ability™ System 800xA® 6.2.pdf, Enclosed Softstarters.pdf, Ex-Solutions.pdf, Low_power_UPS_catalogue_EN.pdf</li>
                     </ul>
                     ''')
                     gr.HTML('</div>')