Create app.py

app.py

@@ -0,0 +1,174 @@
+import gradio as gr
+import base64
+import requests
+import io
+from PIL import Image
+import json
+import os
+from together import Together
+import tempfile
+
+def encode_image_to_base64(image_path):
+    """Convert image to base64 encoding"""
+    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:  # Gradio returns image as dict
+            # Copy the uploaded image to our temporary file
+            with open(image["path"], "rb") as img_file:
+                temp_file.write(img_file.read())
+        elif isinstance(image, Image.Image):
+            # If it's a PIL Image, save it
+            image.save(temp_file.name, format="JPEG")
+        else:
+            # Try to handle other formats
+            try:
+                Image.open(image).save(temp_file.name, format="JPEG")
+            except Exception:
+                return None
+    
+    return temp_file.name
+
+def get_recipe_suggestions(api_key, image, num_recipes=3, dietary_restrictions="None"):
+    """
+    Get recipe suggestions based on the uploaded image of ingredients
+    """
+    if not api_key or not image:
+        return "Please provide both an API key and an image of ingredients."
+    
+    # Save the uploaded image
+    image_path = save_uploaded_image(image)
+    if not image_path:
+        return "Failed to process the uploaded image."
+    
+    try:
+        # Initialize Together client with the provided API key
+        client = Together(api_key=api_key)
+        
+        # Create the prompt for the model
+        system_prompt = """You are a culinary expert AI assistant that specializes in creating recipes based on available ingredients.
+        Analyze the provided image of ingredients and suggest creative, detailed recipes.
+        For each recipe suggestion, include:
+        1. Recipe name
+        2. Brief description of the dish
+        3. Complete ingredients list (including estimated quantities and any additional staple ingredients that might be needed)
+        4. Step-by-step cooking instructions
+        5. Approximate cooking time
+        6. Difficulty level (Easy, Medium, Advanced)
+        7. Nutritional highlights
+        
+        Consider any dietary restrictions mentioned by the user."""
+        
+        user_prompt = f"""Based on the ingredients shown in this image, suggest {num_recipes} creative and delicious recipes.
+        Dietary restrictions to consider: {dietary_restrictions}
+        Please be specific about what ingredients you can identify in the image and creative with your recipe suggestions."""
+        
+        # Create message with image
+        response = client.chat.completions.create(
+            model="meta-llama/Llama-Vision-Free",
+            messages=[
+                {
+                    "role": "system",
+                    "content": system_prompt
+                },
+                {
+                    "role": "user",
+                    "content": [
+                        {
+                            "type": "text",
+                            "text": user_prompt
+                        },
+                        {
+                            "type": "image_url",
+                            "image_url": {
+                                "url": f"file://{image_path}"
+                            }
+                        }
+                    ]
+                }
+            ],
+            max_tokens=2048,
+            temperature=0.7
+        )
+        
+        # Clean up the temporary file
+        try:
+            os.unlink(image_path)
+        except:
+            pass
+        
+        return response.choices[0].message.content
+        
+    except Exception as e:
+        # Clean up the temporary file in case of error
+        try:
+            os.unlink(image_path)
+        except:
+            pass
+        return f"Error: {str(e)}"
+
+# Create the Gradio interface
+with gr.Blocks(title="Visual Recipe Assistant") as app:
+    gr.Markdown("# 🍲 Visual Recipe Assistant")
+    gr.Markdown("Upload an image of ingredients you have, and get creative recipe suggestions!")
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            api_key_input = gr.Textbox(
+                label="Together API Key", 
+                placeholder="Enter your Together API key here...",
+                type="password"
+            )
+            image_input = gr.Image(
+                label="Upload Image of Ingredients",
+                type="filepath"
+            )
+            
+            with gr.Row():
+                num_recipes = gr.Slider(
+                    minimum=1, 
+                    maximum=5, 
+                    value=3, 
+                    step=1, 
+                    label="Number of Recipe Suggestions"
+                )
+                dietary_restrictions = gr.Dropdown(
+                    choices=["None", "Vegetarian", "Vegan", "Gluten-Free", "Dairy-Free", "Low-Carb", "Keto"],
+                    value="None",
+                    label="Dietary Restrictions"
+                )
+            
+            submit_button = gr.Button("Get Recipe Suggestions", variant="primary")
+        
+        with gr.Column(scale=2):
+            output = gr.Markdown(label="Recipe Suggestions")
+    
+    # Set up the submission action
+    submit_button.click(
+        fn=get_recipe_suggestions,
+        inputs=[api_key_input, image_input, num_recipes, dietary_restrictions],
+        outputs=output
+    )
+    
+    gr.Markdown("""
+    ## How to Use
+    1. Enter your Together API key
+    2. Upload an image of ingredients you have available
+    3. Adjust the number of recipes you'd like to receive
+    4. Select any dietary restrictions
+    5. Click "Get Recipe Suggestions"
+    
+    ## About
+    This app uses the Llama-Vision-Free multimodal model from Meta to analyze images of ingredients 
+    and suggest creative recipes based on what it identifies.
+    """)
+
+# Launch the app
+if __name__ == "__main__":
+    app.launch()