Update app.py

app.py

@@ -7,13 +7,13 @@ from PIL import Image
 openai.api_key = os.getenv('OPENAI_API_KEY').strip()
 nutrition_prompt = os.getenv('NUTRITION_PROMPT').strip()
 
-# Function to call GPT-4o-mini (or GPT-3.5/4) to get nutritional information from text description
+# Function to call GPT-4o-mini to get nutritional information from text description
 def get_nutritional_info(description):
     prompt = f"{nutrition_prompt}: {description}"
     
     try:
         response = openai.ChatCompletion.create(
-            model="gpt-4o-mini",  # Assuming GPT-4o-mini is the model you're using (could be gpt-3.5-turbo, etc.)
+            model="gpt-4o-mini",  # Ensure this is the correct model identifier
             messages=[
                 {"role": "system", "content": "You are a nutrition expert."},
                 {"role": "user", "content": prompt}
@@ -28,20 +28,29 @@ def get_nutritional_info(description):
 # Gradio interface function
 def analyze_meal(image, description):
     if image is not None:
-        # Since GPT-4o-mini does not support image input, we ask the user to describe the meal
-        return "Yum! Looks like a delicious dish, but image analysis is not supported. Please describe your meal below!"
+        # Handling if image is given but no text description is provided
+        if not description:
+            return "Please describe your meal in the text box below the image upload."
+        try:
+            # Process the image if necessary or use the description
+            result = get_nutritional_info(description)
+            return result
+        except Exception as e:
+            return f"Error processing image or description: {str(e)}"
     elif description:
         # Use the description to get nutritional information
-        result = get_nutritional_info(description)
-        return result
+        try:
+            result = get_nutritional_info(description)
+            return result
+        except Exception as e:
+            return f"Error processing description: {str(e)}"
     else:
-        return "Nomp Nomp! Please upload an image or provide a description of the meal."
+        return "Please upload an image and provide a description of the meal."
 
-# Gradio app layout (swapped the positions of image upload and text description)
+# Gradio app layout
 inputs = [
-    gr.Image(label="Upload your meal (Take a bite out of that picture!)", type="pil",
-    gr.Textbox(label="Describe your meal if image recognition fails", lines=2, placeholder="e.g., a plate of upma with coconut chutney"),
-    )
+    gr.Image(label="Upload your meal (Take a bite out of that picture!)", type="pil"),
+    gr.Textbox(label="Describe your meal if image recognition fails", lines=2, placeholder="e.g., a plate of upma with coconut chutney")
 ]
 
 outputs = gr.Textbox(label="Nutritional Breakdown (Let's see what’s on your plate!)")