Update app.py

app.py

@@ -2,55 +2,45 @@ import gradio as gr
 from image_processing import preprocess_image
 from food_identification import identify_food
 from nutritional_analysis import get_nutritional_info
-from portion_size_analysis import estimate_portion_size
 
 def analyze_food(image):
+    """
+    Analyze the uploaded image for food items and provide dynamic nutritional information.
+    """
     # Step 1: Preprocess the image
     preprocessed_image = preprocess_image(image)
-    
-    # Step 2: Identify food items
-    food_items = identify_food(preprocessed_image)
-    
-    # Filter non-food items (manually exclude invalid detections)
-    food_items = [item for item in food_items if item not in ["packet", "spoon", "bowl", "plate"]]
-    
-    # Step 3: Fetch nutritional information
-    nutrition_data = get_nutritional_info(food_items)
-    
-    # Step 4: Estimate portion size
-    portion_size = estimate_portion_size(image)
 
-    # Format output
-    if not food_items:
-        return "No valid food items detected. Please try again with a clearer food image."
+    # Step 2: Detect food items dynamically
+    food_item = identify_food(preprocessed_image)
 
-    description = f"Detected {len(food_items)} food items: {', '.join(food_items)}.\n\n"
-    description += f"Portion size: {portion_size}.\n\n"
+    # Step 3: Fetch dynamic nutritional data
+    nutrition_data = get_nutritional_info(food_item)
+
+    # Format the output
+    description = f"Detected food item: {food_item}.\n\n"
     description += "Nutritional Information:\n"
 
-    for food, nutrients in nutrition_data.items():
-        if "Error" in nutrients:
-            description += f"- {food}: {nutrients['Error']}\n"
-        else:
-            description += (
-                f"- {food}:\n"
-                f"  - Energy: {nutrients['Energy (kcal)']} kcal\n"
-                f"  - Protein: {nutrients['Protein (g)']} g\n"
-                f"  - Carbs: {nutrients['Carbs (g)']} g\n"
-                f"  - Fiber: {nutrients['Fiber (g)']} g\n"
-                f"  - Fat: {nutrients['Fat (g)']} g\n"
-                f"  - Sugar: {nutrients['Sugar (g)']} g\n"
-            )
+    if "Error" in nutrition_data:
+        description += f"- {food_item}: {nutrition_data['Error']}\n"
+    else:
+        description += (
+            f"- {food_item}:\n"
+            f"  - Energy: {nutrition_data['Energy (kcal)']} kcal\n"
+            f"  - Protein: {nutrition_data['Protein (g)']} g\n"
+            f"  - Carbs: {nutrition_data['Carbs (g)']} g\n"
+            f"  - Fiber: {nutrition_data['Fiber (g)']} g\n"
+            f"  - Fat: {nutrition_data['Fat (g)']} g\n"
+            f"  - Sugar: {nutrition_data['Sugar (g)']} g\n"
+        )
     return description
 
-
 # Gradio Interface
 iface = gr.Interface(
     fn=analyze_food,
     inputs=gr.Image(type="pil"),
     outputs="text",
     title="Diet Nutrition Analyzer",
-    description="Upload an image of your food plate to analyze nutrition and portion size."
+    description="Upload an image of your food plate to analyze nutrition dynamically."
 )
 
 if __name__ == "__main__":