Create app.py

app.py

@@ -0,0 +1,159 @@
+import gradio as gr
+import requests
+from transformers import pipeline
+
+# Initialize translation models
+translate_to_english = pipeline("translation", model="Helsinki-NLP/opus-mt-et-en")
+translate_to_estonian = pipeline("translation", model="Helsinki-NLP/opus-mt-en-et")
+
+# Spoonacular API settings
+API_KEY = '0bf3de98d2654cb28f139df9fedb7db2'
+SPOONACULAR_URL = "https://api.spoonacular.com/recipes/findByIngredients"
+RECIPE_INFO_URL = "https://api.spoonacular.com/recipes/{}/information"
+RECIPE_SEARCH_URL = "https://api.spoonacular.com/recipes/complexSearch"
+
+def translate_text(input_text, direction='et-en'):
+    """Translate text between Estonian and English."""
+    if direction == 'et-en':
+        result = translate_to_english(input_text)
+    else:
+        result = translate_to_estonian(input_text)
+    return result[0]['translation_text']
+
+def find_recipes(ingredient, calories, gluten_content):
+    """Find recipes based on the ingredient and dietary preferences."""
+    # Translate ingredient to English
+    ingredient_en = translate_text(ingredient, direction='et-en')
+    
+    # Map Estonian options to English for API
+    calories_map = {
+        "Madal kalorsus": "Low",
+        "Kõrge kalorsus": "High"
+    }
+    gluten_map = {
+        "Sisaldab gluteeni": "Contains",
+        "Gluteenivaba": "Free of"
+    }
+    
+    # Search for recipes that include the specific ingredient
+    search_params = {
+        "apiKey": API_KEY,
+        "query": ingredient_en,
+        "includeIngredients": ingredient_en,
+        "number": 5,  # Fetch multiple recipes to increase chances
+    }
+    
+    # Add diet filters
+    if gluten_content == "Gluteenivaba":
+        search_params["intolerances"] = "Gluten"
+    
+    response = requests.get(RECIPE_SEARCH_URL, params=search_params)
+    
+    if response.status_code == 200:
+        recipes = response.json().get('results', [])
+        
+        if not recipes:
+            return "Ühtegi retsepti ei leitud.", ""
+        
+        # Try to find a suitable recipe
+        for recipe in recipes:
+            # Get detailed recipe information
+            recipe_info_response = requests.get(
+                RECIPE_INFO_URL.format(recipe['id']),
+                params={
+                    "apiKey": API_KEY,
+                    "includeNutrition": True
+                }
+            )
+            
+            if recipe_info_response.status_code == 200:
+                recipe_info = recipe_info_response.json()
+                
+                # Verify ingredient is in the recipe
+                ingredients_list = recipe_info.get('extendedIngredients', [])
+                ingredient_names = [ing['name'].lower() for ing in ingredients_list]
+                
+                # Check if the main ingredient is in the recipe
+                if any(ingredient_en.lower() in name for name in ingredient_names):
+                    # Translate recipe title
+                    recipe_name_en = recipe_info['title']
+                    recipe_name_et = translate_text(recipe_name_en, direction='en-et')
+                    
+                    # Create recipe link
+                    recipe_link = recipe_info.get('sourceUrl', 'Link pole saadaval')
+                    
+                    # Translate ingredients to Estonian
+                    ingredients_list = [f"- {ingredient['original']}" for ingredient in ingredients_list]
+                    ingredients_en = '\n'.join(ingredients_list)
+                    ingredients_et = translate_text(ingredients_en, direction='en-et')
+                    
+                    # Translate instructions
+                    instructions_en = recipe_info.get('instructions', 'Instructions not available')
+                    instructions_et = translate_text(instructions_en, direction='en-et')
+                    
+                    # Calculate calories per serving
+                    nutrition = recipe_info.get('nutrition', {})
+                    calories_per_serving = nutrition.get('calories', 'N/A')
+                    
+                    # Create a detailed recipe description
+                    recipe_description = (
+                        f"🍽️ RETSEPT: {recipe_name_et}\n\n"
+                        f"🥘 Põhikoostisosa: {ingredient}\n\n"
+                        f"⏲️ Valmistamisaeg: {recipe_info.get('readyInMinutes', 'N/A')} minutit\n"
+                        f"👥 Portsjonite arv: {recipe_info.get('servings', 'N/A')}\n\n"
+                        f"📝 KOOSTISOSAD:\n{ingredients_et}\n\n"
+                        f"👩‍🍳 VALMISTAMISE JUHISED:\n{instructions_et}\n\n"
+                        f"ℹ️ LISAINFO:\n"
+                        f"Kalorsus: {calories}\n"
+                        f"Gluteenisisaldus: {gluten_content}\n"
+                        f"Kalorid portsjoni kohta: {calories_per_serving}\n\n"
+                        f"🔗 Originaalretsept: {recipe_link}"
+                    )
+                    
+                    return recipe_description, recipe_link
+        
+        # If no recipe found with the exact ingredient
+        return "Kahjuks ei leitud retsepti täpse koostisosaga.", ""
+    else:
+        return "Viga retseptide toomisel.", ""
+
+# Create the Gradio interface with explicit choices
+iface = gr.Interface(
+    fn=find_recipes,
+    inputs=[
+        gr.Textbox(
+            lines=1, 
+            placeholder="Sisestage koostisosa (nt lõhe, šokolaad)", 
+            label="Koostisosa"
+        ),
+        gr.Dropdown(
+            choices=["Madal kalorsus", "Kõrge kalorsus"], 
+            label="Kalorsus",
+            type="value"
+        ),
+        gr.Dropdown(
+            choices=["Sisaldab gluteeni", "Gluteenivaba"], 
+            label="Gluteenisisaldus",
+            type="value"
+        )
+    ],
+    outputs=[
+        gr.Textbox(lines=20, label="Genereeritud Retsept"),
+        gr.Textbox(lines=1, label="Retsepti Link")
+    ],
+    title="🍳 Nutikas Retsepti Generaator",
+    description=(
+        "Sisestage koostisosa, valige kalorsus ja gluteenisisaldus. "
+        "Saate personaalse retsepti!"
+    ),
+    examples=[
+        ["lõhe", "Madal kalorsus", "Gluteenivaba"],
+        ["šokolaadikook", "Kõrge kalorsus", "Sisaldab gluteeni"],
+        ["kartul", "Madal kalorsus", "Gluteenivaba"]
+    ],
+    theme="default"
+)
+
+# Launch the interface
+if __name__ == "__main__":
+    iface.launch()