Update src/streamlit_app.py

src/streamlit_app.py

@@ -197,7 +197,7 @@ Style: Photorealistic, magazine-quality food photography that would be used in h
 
         result = client.images.generate(**gen_params)
 
-        if not result or not result.data or not len(result.data) == 0:
+        if not result or not result.data or len(result.data) == 0:
             st.error(f"No image data returned for {dish_name}")
             return None, None
 
@@ -214,7 +214,7 @@ Style: Photorealistic, magazine-quality food photography that would be used in h
         # Create a data URL for display purpose
         image_url = f"data:image/jpeg;base64,{image_base64}"
 
-        return image_bytes, image_url
+        return image_url, image_bytes
     except Exception as e:
         st.error(f"Error generating food image for {dish_name}: {str(e)}")
         return None, None
@@ -337,7 +337,7 @@ if client:
 
                 status_text.text(f"Generating food photo for {dish_name} ({i+1}/{total_dishes})")
 
-                # Generate food image
+                # When you generate the image
                 image_bytes, image_url = generate_food_image(
                     dish_name,
                     description,
@@ -345,6 +345,13 @@ if client:
                     image_style
                 )
 
+                # Store the image data in the dish_images dictionary
+                if image_bytes and image_url:
+                    dish_images[dish_name] = {
+                        'bytes': image_bytes,
+                        'url': image_url
+                    }
+
                 # Get nutritional information
                 if include_nutrition:
                     nutritional_data[dish_name] = get_nutrinutional_info(dish_name, menu_data.get('cuisine_type', "Fine Dining"))
@@ -353,8 +360,7 @@ if client:
                 progress_bar.progress((i+1)/total_dishes)
 
             progress_bar.progress(1.0)
-            status_text.empty()
-            st.success("Visual menu generation complete!")
+            status_text.text("Visual menu generation complete!")
 
             # Display results
             st.markdown("### Your Visual Menu Results")
@@ -474,7 +480,7 @@ else:
     st.info("Please enter OpenAI key to get started!.")
 
     st.markdown("""
-    ### Trasform you Menu with AI"
+    ### Trasform you Menu with AI
     
     ### **Upload Any Menu Photo**
     - Take a picture of your menu and we'll extract all items automatically!
@@ -484,4 +490,4 @@ else:
     
     ### **Download Your Menu Report**
     - Download detailed report of your menu with all details!    
-    """)
+    """)