Update app.py

app.py

@@ -38,18 +38,7 @@ 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:
@@ -76,6 +65,19 @@ 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