Add application and requirements files

app.py

@@ -1,20 +1,6 @@
-# -*- coding: utf-8 -*-
-"""2000CalRecipeGen_w_KeyIngr_Halal.ipynb
-
-Automatically generated by Colab.
-
-Original file is located at
-    https://colab.research.google.com/drive/1XfF-upk8LT3Gdb4i-yHTRC3X7IFPVrOJ
-"""
-
-# Install dependencies
-#pyttsx3 below does not work; replaced with gTTS
-#!pip install gradio transformers diffusers accelerate torch reportlab pyttsx3 requests --quiet
-
-#below is active line
-#!pip install gradio transformers diffusers accelerate torch reportlab gTTS #requests --quiet
-
+# app.py
 
+from spaces import GPU
 import gradio as gr
 from transformers import pipeline
 from diffusers import StableDiffusionPipeline
@@ -25,8 +11,21 @@ import os
 import json
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
-#import pyttsx3
 from tempfile import NamedTemporaryFile
+from gtts import gTTS
+
+# --- GPU preload hook ---
+@GPU
+def preload():
+    global recipe_model, image_model
+
+    recipe_model = pipeline("text-generation", model="samdak93/qrit-2")
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    image_model = StableDiffusionPipeline.from_pretrained(
+        "runwayml/stable-diffusion-v1-5",
+        torch_dtype=torch.float16 if device == "cuda" else torch.float32
+    ).to(device)
+
 
 # --- USDA API Setup ---
 USDA_API_KEY = "gcwe1gXmMveg7buqddggl6wAZa7Sd7wrZV87P31z"
@@ -50,14 +49,6 @@ def get_calories(ingredient):
     nutrients = get_nutrients(ingredient)
     return nutrients.get("Energy", 0.0)
 
-# --- Models ---
-recipe_model = pipeline("text-generation", model="samdak93/qrit-2")
-device = "cuda" if torch.cuda.is_available() else "cpu"
-image_model = StableDiffusionPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5",
-    torch_dtype=torch.float16 if device == "cuda" else torch.float32
-).to(device)
-
 # --- Utilities ---
 forbidden = ["pork", "bacon", "ham", "lard", "gelatin", "alcohol", "beer", "wine", "rum", "whiskey", "vodka", "gin"]
 
@@ -65,9 +56,9 @@ def parse_ingredients(text):
     match = re.search(r"(?i)ingredients:(.*?)(\n|directions:|instructions:|$)", text, re.DOTALL)
     if match:
         ingredients_block = match.group(1).strip()
-        return [line.strip("- *") for line in ingredients_block.split("\n") if line.strip()]
+        return list(set([line.strip("- *") for line in ingredients_block.split("\n") if line.strip()]))
     return []
-#change 1000 to 2000 calories for more flexibility (2 places only)
+
 def generate_recipe(key_ingredient):
     for _ in range(5):
         prompt = f"Create a halal recipe under 2000 calories that includes {key_ingredient}.\nIngredients:"
@@ -88,6 +79,7 @@ def get_nutrient_breakdown(ingredients):
     return breakdown
 
 def generate_image(prompt):
+    device = "cuda" if torch.cuda.is_available() else "cpu"
     with torch.autocast(device):
         return image_model(prompt).images[0]
 
@@ -120,21 +112,12 @@ def export_pdf(recipe_text, nutrients):
     c.save()
     return temp_pdf.name
 
-#def generate_audio(recipe_text):
-#    engine = pyttsx3.init()
-#    temp_audio = NamedTemporaryFile(delete=False, suffix=".mp3")
-#    engine.save_to_file(recipe_text, temp_audio.name)
-#    engine.runAndWait()
-#    return temp_audio.name
-from gtts import gTTS
-
 def generate_audio(recipe_text):
     tts = gTTS(text=recipe_text)
     temp_audio = NamedTemporaryFile(delete=False, suffix=".mp3")
     tts.save(temp_audio.name)
     return temp_audio.name
 
-
 def app(key_ingredient):
     recipe_text, ingredients, calories = generate_recipe(key_ingredient)
     if not ingredients:
@@ -152,7 +135,7 @@ def app(key_ingredient):
 
 # --- Gradio UI ---
 with gr.Blocks() as demo:
-    gr.Markdown("## 🕌 Halal Recipe Generator - Advanced Edition")
+    gr.Markdown("## 🕌 Halal Recipe Generator - Advanced Edition (GPU-Enabled)")
     with gr.Row():
         key_input = gr.Textbox(label="Key Ingredient")
         submit_btn = gr.Button("Generate Recipe")
@@ -166,5 +149,4 @@ with gr.Blocks() as demo:
     submit_btn.click(app, inputs=key_input,
                      outputs=[output_text, output_img, output_table, output_pdf, output_audio])
 
-#demo.launch(share=True,pwa=True,debug=True)
-demo.launch(share=True)
+demo.launch()