ds

app.py

@@ -4,6 +4,7 @@ from PIL import Image
 from huggingface_hub import InferenceClient
 import os
 import openai  # Added import
+from openai.error import OpenAIError  # For specific exception handling
 
 # Set page configuration
 st.set_page_config(
@@ -17,7 +18,7 @@ def local_css():
     st.markdown(
         """
         <style>
-        /* [Your existing CSS styles here] */
+        /* Your existing CSS styles here */
         </style>
         """, unsafe_allow_html=True
     )
@@ -52,19 +53,23 @@ def get_ingredients_qwen(food_name):
         completion = client.chat.completions.create(
             model="Qwen/Qwen2.5-Coder-32B-Instruct", messages=messages, max_tokens=50
         )
-        generated_text = completion.choices[0].message["content"].strip()
+        generated_text = completion.choices[0]['message']['content'].strip()
         return generated_text
     except Exception as e:
         return f"Error generating ingredients: {e}"
 
 # **Set OpenAI API Key**
-openai.api_key = st.secrets["openai"] # Ensure you have this in your secrets
+openai.api_key = st.secrets["openai"]  # Ensure you have this in your secrets
 
 # Main content
 st.markdown('<div class="title"><h1>DelishAI - Your Culinary Assistant</h1></div>', unsafe_allow_html=True)
 
-# Add banner image
-st.image("IR_IMAGE.png", use_container_width=True)
+# Add banner image with existence check
+banner_image_path = "IR_IMAGE.png"
+if os.path.exists(banner_image_path):
+    st.image(banner_image_path, use_container_width=True)
+else:
+    st.warning(f"Banner image '{banner_image_path}' not found.")
 
 # Sidebar for model information (hidden on small screens)
 with st.sidebar:
@@ -76,13 +81,19 @@ with st.sidebar:
     st.markdown("---")
     st.markdown("<p style='text-align: center;'>Developed by Muhammad Hassan Butt.</p>", unsafe_allow_html=True)
 
-# Assuming `sample_images` is defined somewhere in your code
+# Define sample images with correct paths
 sample_images = {
-    "Pizza": "path_to_pizza_image.jpg",
-    "Salad": "path_to_salad_image.jpg",
+    "Pizza": "sample_images/pizza.jpg",
+    "Salad": "sample_images/salad.jpg",
     # Add more sample images as needed
 }
 
+# Ensure sample images exist
+for name, path in sample_images.items():
+    if not os.path.exists(path):
+        st.warning(f"Sample image '{path}' for '{name}' not found.")
+
+# Create columns for sample image buttons
 cols = st.columns(len(sample_images))
 for idx, (name, file_path) in enumerate(sample_images.items()):
     with cols[idx]:
@@ -92,52 +103,69 @@ for idx, (name, file_path) in enumerate(sample_images.items()):
 # File uploader
 st.subheader("Upload a food image:")
 uploaded_file = st.file_uploader("", type=["jpg", "png", "jpeg"])
-if 'uploaded_file' in locals() and uploaded_file is not None:
+
+if uploaded_file is not None:
     # Display the uploaded image
     if isinstance(uploaded_file, str):  # Sample image selected
-        image = Image.open(uploaded_file)
+        if os.path.exists(uploaded_file):
+            image = Image.open(uploaded_file)
+        else:
+            st.error(f"Sample image '{uploaded_file}' not found.")
+            image = None
     else:  # User uploaded image
         image = Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image", use_container_width=True)
-
-    # Classification button
-    if st.button("Classify"):
-        with st.spinner("Classifying..."):
-            # Make predictions
-            predictions = pipe_classification(image)
-            # Display only the top prediction
-            top_food = predictions[0]['label']
-            st.header(f"🍽️ Food: {top_food}")
-           
-            # Generate and display ingredients for the top prediction
-            st.subheader("📝 Ingredients")
-            try:
-                ingredients = get_ingredients_qwen(top_food)
-                st.write(ingredients)
-            except Exception as e:
-                st.error(f"Error generating ingredients: {e}")
-
-            # **Healthier Alternatives using OpenAI API**
-            st.subheader("💡 Healthier Alternatives")
-            try:
-                response = openai.ChatCompletion.create(
-                    model="gpt-4",  # You can choose the model you prefer
-                    messages=[
-                        {
-                            "role": "system",
-                            "content": "You are a helpful assistant specializing in providing healthy alternatives to various dishes."
-                        },
-                        {
-                            "role": "user",
-                            "content": f"What's a healthy {top_food} recipe, and why is it healthy?"
-                        }
-                    ],
-                    max_tokens=200,  # Adjust as needed
-                    temperature=0.7,  # Adjust creativity level as needed
-                )
-                result = response.choices[0].message.content.strip()
-                st.write(result)
-            except Exception as e:
-                st.error(f"Unable to generate healthier alternatives: {e}")
+
+    if image:
+        st.image(image, caption="Uploaded Image", use_container_width=True)
+
+        # Classification button
+        if st.button("Classify"):
+            with st.spinner("Classifying..."):
+                try:
+                    # Make predictions
+                    predictions = pipe_classification(image)
+                    if predictions:
+                        # Display only the top prediction
+                        top_food = predictions[0]['label']
+                        confidence = predictions[0]['score']
+                        st.header(f"🍽️ Food: {top_food} ({confidence*100:.2f}% confidence)")
+
+                        # Generate and display ingredients for the top prediction
+                        st.subheader("📝 Ingredients")
+                        try:
+                            ingredients = get_ingredients_qwen(top_food)
+                            st.write(ingredients)
+                        except Exception as e:
+                            st.error(f"Error generating ingredients: {e}")
+
+                        # **Healthier Alternatives using OpenAI API**
+                        st.subheader("💡 Healthier Alternatives")
+                        try:
+                            response = openai.ChatCompletion.create(
+                                model="gpt-4",  # You can choose the model you prefer
+                                messages=[
+                                    {
+                                        "role": "system",
+                                        "content": "You are a helpful assistant specializing in providing healthy alternatives to various dishes."
+                                    },
+                                    {
+                                        "role": "user",
+                                        "content": f"What's a healthy {top_food} recipe, and why is it healthy?"
+                                    }
+                                ],
+                                max_tokens=200,  # Adjust as needed
+                                temperature=0.7,  # Adjust creativity level as needed
+                            )
+                            # Corrected access to 'content'
+                            result = response['choices'][0]['message']['content'].strip()
+                            st.write(result)
+                        except OpenAIError as e:
+                            st.error(f"OpenAI API error: {e}")
+                        except Exception as e:
+                            st.error(f"Unable to generate healthier alternatives: {e}")
+                    else:
+                        st.error("No predictions returned from the classification model.")
+                except Exception as e:
+                    st.error(f"Error during classification: {e}")
 else:
     st.info("Please select or upload an image to get started.")

requirements.txt

@@ -4,4 +4,4 @@ torch
 Pillow
 huggingface_hub
 gradio_client
-openai>=1.0.0
+openai==0.28.0