Update app.py

app.py

@@ -14,25 +14,6 @@ def encode_image_to_base64(image_path):
     with open(image_path, "rb") as image_file:
         return base64.b64encode(image_file.read()).decode('utf-8')
 
-def save_uploaded_image(image):
-    """Save uploaded image to a temporary file and return the path"""
-    if image is None:
-        return None
-    
-    with tempfile.NamedTemporaryFile(delete=False, suffix='.jpg') as temp_file:
-        if isinstance(image, dict) and "path" in image:
-            with open(image["path"], "rb") as img_file:
-                temp_file.write(img_file.read())
-        elif isinstance(image, Image.Image):
-            image.save(temp_file.name, format="JPEG")
-        else:
-            try:
-                Image.open(image).save(temp_file.name, format="JPEG")
-            except Exception:
-                return None
-    
-    return temp_file.name
-
 def analyze_single_image(client, img_path):
     """Analyze a single image to identify ingredients"""
     system_prompt = """You are a culinary expert AI assistant that specializes in identifying ingredients in images.
@@ -69,28 +50,14 @@ def analyze_single_image(client, img_path):
     except Exception as e:
         return f"Error analyzing image: {str(e)}"
 
-def get_recipe_suggestions(api_key, images, num_recipes=3, dietary_restrictions="None", cuisine_preference="Any"):
+def get_recipe_suggestions(api_key, image_paths, num_recipes=3, dietary_restrictions="None", cuisine_preference="Any"):
     """Get recipe suggestions based on the uploaded images of ingredients"""
     if not api_key:
         return "Please provide your Together API key."
     
-    if not images or len(images) == 0 or all(img is None for img in images):
+    if not image_paths or len(image_paths) == 0:
         return "Please upload at least one image of ingredients."
     
-    valid_images = [img for img in images if img is not None]
-    
-    if len(valid_images) == 0:
-        return "No valid images were uploaded. Please try again."
-    
-    image_paths = []
-    for img in valid_images:
-        img_path = save_uploaded_image(img)
-        if img_path:
-            image_paths.append(img_path)
-    
-    if not image_paths:
-        return "Failed to process the uploaded images."
-    
     try:
         client = Together(api_key=api_key)
         
@@ -136,12 +103,6 @@ def get_recipe_suggestions(api_key, images, num_recipes=3, dietary_restrictions=
             temperature=0.7
         )
         
-        for img_path in image_paths:
-            try:
-                os.unlink(img_path)
-            except:
-                pass
-        
         result = "## 📋 Ingredients Identified\n\n"
         result += combined_ingredients
         result += "\n\n---\n\n"
@@ -149,15 +110,21 @@ def get_recipe_suggestions(api_key, images, num_recipes=3, dietary_restrictions=
         result += response.choices[0].message.content
         
         return result
-        
     except Exception as e:
-        for img_path in image_paths:
-            try:
-                os.unlink(img_path)
-            except:
-                pass
         return f"Error: {str(e)}"
 
+def update_gallery(files):
+    """Update the gallery with uploaded image paths"""
+    if not files or len(files) == 0:
+        return gr.update(visible=False)
+    return gr.update(value=files, visible=True)
+
+def process_recipe_request(api_key, files, num_recipes, dietary_restrictions, cuisine_preference):
+    """Process the recipe request with uploaded files"""
+    if not files:
+        return "Please upload at least one image of ingredients."
+    return get_recipe_suggestions(api_key, files, num_recipes, dietary_restrictions, cuisine_preference)
+
 custom_css = """
 @import url('https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;500;600;700&display=swap');
 :root {
@@ -472,19 +439,8 @@ with gr.Blocks(css=custom_css) as app:
     
     gr.HTML(html_footer)
     
-    def update_gallery(files):
-        if not files:
-            return gr.Gallery.update(visible=False)
-        return gr.Gallery.update(value=[file.name for file in files], visible=True)
-    
     file_upload.change(fn=update_gallery, inputs=file_upload, outputs=image_input)
     
-    def process_recipe_request(api_key, files, num_recipes, dietary_restrictions, cuisine_preference):
-        if not files:
-            return "Please upload at least one image of ingredients."
-        images = [file.name for file in files]
-        return get_recipe_suggestions(api_key, images, num_recipes, dietary_restrictions, cuisine_preference)
-    
     submit_button.click(
         fn=process_recipe_request,
         inputs=[api_key_input, file_upload, num_recipes, dietary_restrictions, cuisine_preference],