Update app.py

app.py

@@ -1,50 +1,103 @@
 
 
 import gradio as gr
+import requests
 from transformers import pipeline
-import torch
 
-# Initialize the text generation pipeline
-generator = pipeline('text-generation', model='gpt2')
+# 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")
 
-def generate_recipe(ingredient):
-    """
-    Generate a recipe based on the input ingredient/word
-    """
-    # Create a prompt template
-    prompt = f"Recipe for {ingredient}:\n\nIngredients:\n"
+# Spoonacular API settings
+API_KEY = '0bf3de98d2654cb28f139df9fedb7db2'
+SPOONACULAR_URL = "https://api.spoonacular.com/recipes/findByIngredients"
+
+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, wheat_content):
+    """Find recipes based on the ingredient and dietary preferences."""
+    # Translate ingredient to English
+    ingredient_en = translate_text(ingredient, direction='et-en')
     
-    # Generate the recipe
-    try:
-        result = generator(
-            prompt,
-            max_length=300,
-            num_return_sequences=1,
-            temperature=0.7,
-            pad_token_id=generator.tokenizer.eos_token_id
-        )
+    params = {
+        "ingredients": ingredient_en,
+        "apiKey": API_KEY
+    }
+    
+    response = requests.get(SPOONACULAR_URL, params=params)
+    
+    if response.status_code == 200:
+        recipes = response.json()
         
-        # Extract and format the generated recipe
-        recipe = result[0]['generated_text']
+        if not recipes:
+            return "Ühtegi retsepti ei leitud.", ""
         
-        # Clean up the output
-        recipe = recipe.replace(prompt, '')
+        # Select the first recipe for demonstration
+        recipe = recipes[0]
         
-        return recipe
-    
-    except Exception as e:
-        return f"Error generating recipe: {str(e)}"
+        # Prepare recipe details
+        recipe_name_en = recipe['title']
+        recipe_name_et = translate_text(recipe_name_en, direction='en-et')
+        recipe_link = f"https://spoonacular.com/recipes/{recipe['id']}-{recipe_name_en.replace(' ', '-')}"
+        
+        # Translate ingredients to Estonian
+        missed_ingredients_en = ', '.join([ingredient['name'] for ingredient in recipe['missedIngredients']])
+        missed_ingredients_et = translate_text(missed_ingredients_en, direction='en-et')
+        
+        # Create a recipe description
+        recipe_description = (
+            f"Retsept: {recipe_name_et}\n\n"
+            f"Link: {recipe_link}\n\n"
+            f"Koostisosad: {missed_ingredients_et}\n\n"
+            f"Valmistamise juhised: [Link to instructions]"
+        )
+        
+        return recipe_description, recipe_link
+    else:
+        return "Viga retseptide toomisel.", ""
 
 # Create the Gradio interface
 iface = gr.Interface(
-    fn=generate_recipe,
-    inputs=gr.Textbox(lines=1, placeholder="Enter an ingredient or dish name..."),
-    outputs=gr.Textbox(lines=10, label="Generated Recipe"),
-    title="Recipe Generator",
-    description="Enter an ingredient or dish name, and I'll generate a recipe for you!",
-    examples=[["chocolate cake"], ["chicken curry"], ["pasta"]]
+    fn=find_recipes,
+    inputs=[
+        gr.Textbox(
+            lines=1, 
+            placeholder="Sisestage koostisosa (nt lõhe, šokolaad)", 
+            label="Koostisosa"
+        ),
+        gr.Radio(
+            ["Low Calories", "High Calories"], 
+            label="Kalorsus"
+        ),
+        gr.Radio(
+            ["Contains Wheat", "Wheat Free"], 
+            label="Nisusisaldus"
+        )
+    ],
+    outputs=[
+        gr.Textbox(lines=15, label="Genereeritud Retsept"),
+        gr.Textbox(lines=1, label="Retsepti Link")
+    ],
+    title="🍳 Nutikas Retsepti Generator",
+    description=(
+        "Sisestage koostisosa, valige kalorsus ja nisusisaldus. "
+        "Saate personaalse retsepti!"
+    ),
+    examples=[
+        ["lõhe", "Low Calories", "Wheat Free"],
+        ["šokolaadikook", "High Calories", "Contains Wheat"],
+        ["kartul", "Low Calories", "Wheat Free"]
+    ],
+    theme="default"
 )
 
 # Launch the interface
 if __name__ == "__main__":
     iface.launch()
+