Update app.py

app.py

@@ -38,6 +38,19 @@ def process_audio(audio_path, image_option, creative_text_option):
     if audio_path is None:
         return "Please upload an audio file.", None, None, None
 
+    # Step 4: Generate image (if selected)
+    image = None
+    if image_option == "Generate Image":
+        try:
+            model_id1 = "dreamlike-art/dreamlike-diffusion-1.0"
+            pipe = StableDiffusionPipeline.from_pretrained(model_id1, torch_dtype=torch.float16, use_safetensors=True)
+            pipe = pipe.to('cpu')
+            image = pipe(translation).images[0]
+        except Exception as e:
+            return tamil_text, translation, creative_text, f"An error occurred during image generation: {str(e)}"
+
+    return tamil_text, translation, creative_text, image    
+
     # Step 1: Transcribe audio
     try:
         with open(audio_path, "rb") as file:
@@ -63,18 +76,7 @@ def process_audio(audio_path, image_option, creative_text_option):
     if creative_text_option == "Generate Creative Text":
         creative_text = generate_creative_text(translation)
 
-    # Step 4: Generate image (if selected)
-    image = None
-    if image_option == "Generate Image":
-        try:
-            model_id1 = "dreamlike-art/dreamlike-diffusion-1.0"
-            pipe = StableDiffusionPipeline.from_pretrained(model_id1, torch_dtype=torch.float16, use_safetensors=True)
-            pipe = pipe.to('cpu')
-            image = pipe(translation).images[0]
-        except Exception as e:
-            return tamil_text, translation, creative_text, f"An error occurred during image generation: {str(e)}"
-
-    return tamil_text, translation, creative_text, image
+    
 
 # Create Gradio interface
 with gr.Blocks(theme=gr.themes.Base()) as iface: