Update src/streamlit_app.py

src/streamlit_app.py

@@ -187,16 +187,18 @@ Technical Requirements:
 Style: Photorealistic, magazine-quality food photography that would be used in high-end restaurant menu or food advertising.
 """
     try:
-        result = client.images.generate(
-            model="gpt-image-1",
-            prompt=food_prompt,
-            size="1024x1024",
-            quality="high",
-            n=1
-        )
+        gen_params = {
+            "model": "gpt-image-1",
+            "prompt": food_prompt,
+            "size": "1024x1024",
+            "quality": "high",
+            "n": 1
+        }
+
+        result = client.images.generate(**gen_params)
 
         if not result or not result.data or not len(result.data) == 0:
-            st.error(f"""No image data returned for {dish_name}""")
+            st.error(f"No image data returned for {dish_name}")
             return None, None
 
         # Get base64 image data directly ( it should work directly )
@@ -204,7 +206,7 @@ Style: Photorealistic, magazine-quality food photography that would be used in h
 
         # Check if base64 data is valid
         if not image_base64:
-            st.error(f"""No base64 image returned for {dish_name}""")
+            st.error(f"No base64 image returned for {dish_name}")
             return None, None
 
         image_bytes = base64.b64decode(image_base64)
@@ -378,12 +380,10 @@ if client:
 
                         if include_nutrition and dish_name in nutritional_data:
                             nutrition = nutritional_data[dish_name]
-                            st.markdown(
-                                f"**Nutrition** {nutrition.get('calories', 'N/A')} cal, {nutrition.get('protein', 'N/A')} protein"
-                            )
+                            st.markdown(f"**Nutrition** {nutrition.get('calories', 'N/A')} cal, {nutrition.get('protein', 'N/A')} protein")
 
                         if item.get('description'):
-                            st.markdown(f"  *{item['description']}*")
+                            st.markdown(f"*{item['description']}*")
 
 
         # Download Options
@@ -420,8 +420,8 @@ if client:
         col_download1, col_download2 = st.columns(2)
 
         with col_download1:
-            st.markdown("**Bulk Download**")
-            st.info("Individual photos available above, or use menu report for compelete details")
+            st.markdown(f"**Bulk Download**")
+            st.info(f"Individual photos available above, or use menu report for compelete details")
 
         with col_download2:
             menu_summary = f"""VISUAL MENU GENERATED REPORT
@@ -442,7 +442,7 @@ Menu Items Generated: {total_dishes}
                     menu_summary += f"- {item['name']} - {item.get('price', 'N/A')}\n"
                     if include_nutrition and item['name'] in nutritional_data:
                         nutrition = nutritional_data[item['name']]
-                        menu_summary += f"  Nutrition: {nutrition.get('calories', 'N/A')} cal, {nutrition.get('protein', 'N/A')} protein\n"
+                        menu_summary += f"Nutrition: {nutrition.get('calories', 'N/A')} cal, {nutrition.get('protein', 'N/A')} protein\n"
 
             menu_summary += f"""
 Generation Statistics: