Update app.py

app.py

@@ -1,18 +1,16 @@
 import requests
 import io
-from PIL import Image
+from PIL import Image, UnidentifiedImageError
 import gradio as gr
 from transformers import MarianMTModel, MarianTokenizer
 import os
 
-# Model and language settings
+
 model_name = "Helsinki-NLP/opus-mt-mul-en"
 model = MarianMTModel.from_pretrained(model_name)
 tokenizer = MarianTokenizer.from_pretrained(model_name)
 
-
 def translate_text(input_text, language):
-    
     language_map = {
         "Tamil": "ta",
         "French": "fr",
@@ -20,12 +18,10 @@ def translate_text(input_text, language):
         "German": "de"
     }
     
-    
-    lang_prefix = f">>{language_map[language]}<<"
+    lang_prefix = f">>{language_map[language]}<< "  
     text_with_lang = lang_prefix + input_text
     inputs = tokenizer(text_with_lang, return_tensors="pt", padding=True)
-    
-    
+
     translated_tokens = model.generate(**inputs)
     translation = tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
     return translation
@@ -56,27 +52,34 @@ def query_image(payload):
     return response.content
 
 
-def process_input(input_text, language):
+def process_input(tamil_input, language):
     gemini_api_key = os.getenv('GEMINI_API_KEY')  
-    translated_output = translate_text(input_text, language)  
-    creative_output = query_gemini_api(translated_output, gemini_api_key)  
-    image_bytes = query_image({"inputs": translated_output})  
-    image = Image.open(io.BytesIO(image_bytes))  
-    
-    return translated_output, creative_output, image
+    translated_output = translate_text(tamil_input, language)
+    creative_output = query_gemini_api(translated_output, gemini_api_key)
+    image_bytes = query_image({"inputs": translated_output})
+
+    try:
+        image = Image.open(io.BytesIO(image_bytes))
+    except UnidentifiedImageError:
+        image = None  
 
+    return translated_output, creative_output, image
 
+# Gradio interface setup
 iface = gr.Interface(
     fn=process_input,
-    inputs=[gr.Textbox(label="Enter text in the selected language"),
-            gr.Dropdown(choices=["Tamil", "French", "Hindi", "German"], label="Select Source Language")],
+    inputs=[
+        gr.Textbox(label="Input Text"),
+        gr.Dropdown(label="Select Language", choices=["Tamil", "French", "Hindi", "German"])
+    ],
     outputs=[
-        gr.Textbox(label="Translated Text (English)"),
+        gr.Textbox(label="Translated Text"),
         gr.Textbox(label="Creative Text"),
         gr.Image(label="Generated Image")
     ],
-    title="TRANSART🎨 by Sakthi",
-    description="Select a language, enter text, and get the English translation along with creative content and image."
+    title="TRANSART🎨 BY Sakthi",
+    description="Enter text to translate into English and generate an image based on the translated text."
 )
 
 iface.launch()
+