Update app.py

app.py

@@ -18,12 +18,20 @@ translator_tokenizer = MarianTokenizer.from_pretrained(translator_model)
 generator_model = "EleutherAI/gpt-neo-1.3B"
 generator = AutoModelForCausalLM.from_pretrained(generator_model).to(device)
 generator_tokenizer = AutoTokenizer.from_pretrained(generator_model)
+
+# Ensure tokenizer has a padding token
 if generator_tokenizer.pad_token is None:
     generator_tokenizer.pad_token = generator_tokenizer.eos_token
 
-# Hugging Face API for Image Generation
+# Set Hugging Face API Key
 HF_API_KEY = os.getenv("HF_API_KEY")  # Use environment variable
-IMAGE_GEN_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
+if not HF_API_KEY:
+    print("⚠️ Hugging Face API key is missing! Set HF_API_KEY in your environment.")
+else:
+    print("✅ Hugging Face API key detected.")
+
+# Use Stable Diffusion Model for Image Generation
+IMAGE_GEN_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
 HEADERS = {"Authorization": f"Bearer {HF_API_KEY}"}
 
 def translate_tamil_to_english(text):
@@ -41,15 +49,22 @@ def generate_text(prompt):
 def generate_image(prompt):
     """Sends request to API for image generation."""
     response = requests.post(IMAGE_GEN_URL, headers=HEADERS, json={"inputs": prompt})
+    
+    print("🔄 Image Generation Request Sent!")
+    print(f"Status Code: {response.status_code}")
+    
     if response.status_code == 200:
+        print("✅ Image successfully generated!")
         return Image.open(io.BytesIO(response.content))
+    
+    print("❌ Error in image generation:", response.text)
     return Image.new("RGB", (300, 300), "red")  # Error placeholder image
 
 def process_input(tamil_text):
     """Complete pipeline: Translation, Text Generation, and Image Generation."""
     english_text = translate_tamil_to_english(tamil_text)
     creative_text = generate_text(english_text)
-    image = generate_image(english_text)
+    image = generate_image(creative_text)
     return english_text, creative_text, image
 
 # Create Gradio Interface
@@ -67,3 +82,4 @@ interface = gr.Interface(
 
 # Launch the Gradio app
 interface.launch()
+