Update app.py

app.py

@@ -34,32 +34,37 @@ generator.config.pad_token_id = generator_tokenizer.pad_token_id
 
 def translate_tamil_to_english(text):
     """Translate Tamil text to English."""
-    inputs = translator_tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=128).to(device)
-    output = translator.generate(**inputs)
-    return translator_tokenizer.decode(output[0], skip_special_tokens=True)
+    try:
+        inputs = translator_tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=128).to(device)
+        output = translator.generate(**inputs)
+        return translator_tokenizer.decode(output[0], skip_special_tokens=True)
+    except Exception as e:
+        return f"Translation error: {str(e)}"
 
 def generate_text(prompt):
     """Generate creative text from English input."""
-    inputs = generator_tokenizer(prompt, return_tensors="pt", padding=True, truncation=True, max_length=128).to(device)
-    output = generator.generate(**inputs, max_length=100)
-    return generator_tokenizer.decode(output[0], skip_special_tokens=True)
+    try:
+        inputs = generator_tokenizer(prompt, return_tensors="pt", padding=True, truncation=True, max_length=128).to(device)
+        output = generator.generate(**inputs, max_length=100)
+        return generator_tokenizer.decode(output[0], skip_special_tokens=True)
+    except Exception as e:
+        return f"Text generation error: {str(e)}"
 
 def generate_image(prompt):
     """Generate image using Hugging Face inference API."""
-    response = requests.post(IMAGE_GEN_URL, headers=HEADERS, json={"inputs": prompt})
-    if response.status_code == 200:
-        # Check if raw image or base64 encoded
-        content_type = response.headers.get("content-type", "")
-        if "image" in content_type:
-            return Image.open(io.BytesIO(response.content))
-        else:
-            try:
+    try:
+        response = requests.post(IMAGE_GEN_URL, headers=HEADERS, json={"inputs": prompt})
+        if response.status_code == 200:
+            content_type = response.headers.get("content-type", "")
+            if "image" in content_type:
+                return Image.open(io.BytesIO(response.content))
+            else:
                 result = response.json()
                 if "image" in result:
                     image_data = base64.b64decode(result["image"])
                     return Image.open(io.BytesIO(image_data))
-            except Exception as e:
-                print("Error parsing image:", e)
+    except Exception as e:
+        print("Error generating image:", e)
     return Image.new("RGB", (300, 300), "red")  # fallback placeholder
 
 def process_input(tamil_text):
@@ -83,4 +88,4 @@ interface = gr.Interface(
 )
 
 # Launch app
-interface.launch()
+interface.launch()