Update app.py

app.py

@@ -29,19 +29,12 @@ def query_image_generation(payload, max_retries=5):
         return response.content
     return None
 
+
 # Function for generating an image from text
 def generate_image(prompt):
     image_bytes = query_image_generation({"inputs": prompt})
-
-    if image_bytes is None:
-        return None
-
-    try:
-        image = Image.open(io.BytesIO(image_bytes))  # Opening the image from bytes
-        return image
-    except Exception as e:
-        print(f"Error: {e}")
-        return None
+    image = Image.open(io.BytesIO(image_bytes))  # Opening the image from bytes
+    return image
 
 
 # Updated function for text generation using the new API structure
@@ -97,11 +90,11 @@ def process_audio(audio_path, image_option, creative_text_option):
 
 # Create Gradio interface
 with gr.Blocks(theme=gr.themes.Base()) as iface:
-    gr.Markdown("# Audio Transcription, Translation, Image & Creative Text Generation")
+    gr.Markdown("# Multimodal Application for Vernacular Language Translation and Image Synthesis")
     with gr.Row():
         with gr.Column():
             audio_input = gr.Audio(type="filepath", label="Upload Audio File")
-            image_option = gr.Dropdown(["Generate Image", "Skip Image"], label="Image Generation", value="Generate Image")
+            image_option = gr.Dropdown(["Generate Image", "Skip Image"], label="Image Generation", value="Click here to generate")
             creative_text_option = gr.Dropdown(["Generate Creative Text", "Skip Creative Text"], label="Creative Text Generation", value="Generate Creative Text")
             submit_button = gr.Button("Process Audio")
         with gr.Column():