Update app.py

app.py

@@ -1,3 +1,4 @@
+# hugging face
 import gradio as gr
 import requests
 from transformers import MarianMTModel, MarianTokenizer, AutoModelForCausalLM, AutoTokenizer
@@ -18,19 +19,11 @@ 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
 
-# Set Hugging Face API Key
+# Hugging Face API for Image Generation
 HF_API_KEY = os.getenv("HF_API_KEY")  # Use environment variable
-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/black-forest-labs/FLUX.1-schnell"
 HEADERS = {"Authorization": f"Bearer {HF_API_KEY}"}
 
@@ -49,22 +42,15 @@ 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(creative_text)
+    image = generate_image(english_text)
     return english_text, creative_text, image
 
 # Create Gradio Interface
@@ -81,5 +67,4 @@ interface = gr.Interface(
 )
 
 # Launch the Gradio app
-interface.launch()
-
+interface.launch()