Update app.py

app.py

@@ -129,14 +129,19 @@ def change_video_codec(input_file, output_file, codec='libx264', audio_codec='aa
 
 def get_talk(image_in, speech):
     client = Client("fffiloni/dreamtalk")
-    result = client.predict(
-        audio_input=handle_file(speech),
-        image_path=handle_file(image_in),
-        emotional_style="M030_front_neutral_level1_001.mat",
-        api_name="/infer"
-    )
-    print(result)
-    return result['video']
+    try:
+        result = client.predict(
+            audio_input=handle_file(speech),
+            image_path=handle_file(image_in),
+            emotional_style="M030_front_neutral_level1_001.mat",
+            api_name="/infer"
+        )
+        print(result)  # Debugging line
+        return result['video']
+    except Exception as e:
+        print(f"Error in get_talk: {e}")
+        raise gr.Error('An error occurred while loading DreamTalk: Image may not contain any face')
+
 
 #######################################################
 # Gradio APIs for optional image and voice generation #
@@ -212,15 +217,15 @@ def get_whisperspeech(prompt_audio_whisperspeech, audio_to_clone):
 
 
 
-def pipe (voice, image_in):
-    
+def pipe(voice, image_in):
     try:
         video = get_talk(image_in, voice)
-    except:
-       
-        raise gr.Error('An error occurred while loading DreamTalk: Image may not contain any face')
+        print(f"Generated video: {video}")  # Debugging line
+        return video
+    except Exception as e:
+        print(f"Error in pipe: {e}")
+        raise gr.Error('An error occurred while generating the talking portrait.')
 
-    return video
  
     
 
@@ -455,10 +460,12 @@ with gr.Blocks(css=css) as demo:
     )
 
     submit_btn.click(
-        fn = pipe,
-        inputs = [portrait, preprocess_audio_file],
-        outputs = [result],
-        show_api = False
+    fn=pipe,
+    inputs=[portrait, preprocess_audio_file],
+    outputs=[result],
+    show_api=False
+
+
     )