Update app.py

app.py

@@ -15,11 +15,11 @@ from gradio_client import Client
 def get_speech(text, voice):
     client = Client("https://collabora-whisperspeech.hf.space/")
     result = client.predict(
-    		text,	# str  in 'Enter multilingual text💬📝' Textbox component
-    		voice,	# filepath  in 'Upload or Record Speaker Audio (optional)🌬️💬' Audio component
-    		"",	# str  in 'alternatively, you can paste in an audio file URL:' Textbox component
-    		14,	# float (numeric value between 10 and 15) in 'Tempo (in characters per second)' Slider component
-    		api_name="/whisper_speech_demo"
+      text, # str  in 'Enter multilingual text💬📝' Textbox component
+      voice, # filepath  in 'Upload or Record Speaker Audio (optional)🌬️💬' Audio component
+      "", # str  in 'alternatively, you can paste in an audio file URL:' Textbox component
+      14, # float (numeric value between 10 and 15) in 'Tempo (in characters per second)' Slider component
+      api_name="/whisper_speech_demo"
     )
     print(result)
     return result
@@ -27,15 +27,15 @@ def get_speech(text, voice):
 def get_dreamtalk(image_in, speech):
     client = Client("https://fffiloni-dreamtalk.hf.space/")
     result = client.predict(
-    		speech,	# filepath  in 'Audio input' Audio component
-    		image_in,	# filepath  in 'Image' Image component
-    		"M030_front_neutral_level1_001.mat",	# Literal['M030_front_angry_level3_001.mat', 'M030_front_contempt_level3_001.mat', 'M030_front_disgusted_level3_001.mat', 'M030_front_fear_level3_001.mat', 'M030_front_happy_level3_001.mat', 'M030_front_neutral_level1_001.mat', 'M030_front_sad_level3_001.mat', 'M030_front_surprised_level3_001.mat', 'W009_front_angry_level3_001.mat', 'W009_front_contempt_level3_001.mat', 'W009_front_disgusted_level3_001.mat', 'W009_front_fear_level3_001.mat', 'W009_front_happy_level3_001.mat', 'W009_front_neutral_level1_001.mat', 'W009_front_sad_level3_001.mat', 'W009_front_surprised_level3_001.mat', 'W011_front_angry_level3_001.mat', 'W011_front_contempt_level3_001.mat', 'W011_front_disgusted_level3_001.mat', 'W011_front_fear_level3_001.mat', 'W011_front_happy_level3_001.mat', 'W011_front_neutral_level1_001.mat', 'W011_front_sad_level3_001.mat', 'W011_front_surprised_level3_001.mat']  in 'emotional style' Dropdown component
-    		api_name="/infer"
+      speech, # filepath  in 'Audio input' Audio component
+      image_in, # filepath  in 'Image' Image component
+      "M030_front_neutral_level1_001.mat", # Literal['M030_front_angry_level3_001.mat', 'M030_front_contempt_level3_001.mat', 'M030_front_disgusted_level3_001.mat', 'M030_front_fear_level3_001.mat', 'M030_front_happy_level3_001.mat', 'M030_front_neutral_level1_001.mat', 'M030_front_sad_level3_001.mat', 'M030_front_surprised_level3_001.mat', 'W009_front_angry_level3_001.mat', 'W009_front_contempt_level3_001.mat', 'W009_front_disgusted_level3_001.mat', 'W009_front_fear_level3_001.mat', 'W009_front_happy_level3_001.mat', 'W009_front_neutral_level1_001.mat', 'W009_front_sad_level3_001.mat', 'W009_front_surprised_level3_001.mat', 'W011_front_angry_level3_001.mat', 'W011_front_contempt_level3_001.mat', 'W011_front_disgusted_level3_001.mat', 'W011_front_fear_level3_001.mat', 'W011_front_happy_level3_001.mat', 'W011_front_neutral_level1_001.mat', 'W011_front_sad_level3_001.mat', 'W011_front_surprised_level3_001.mat']  in 'emotional style' Dropdown component
+      api_name="/infer"
     )
     print(result)
     return result['video']
 
-st.title("✨️ AI Partne")
+st.title("✨️ AI Partner")
 st.header("To create partner use sidebar (tap on ▶️ button on the left top)", divider="rainbow")
 st.sidebar.title("✨ Your AI Girl 🔞")
 st.sidebar.write("We will make a perfect talking girlfriend for you, just upload photo")
@@ -52,21 +52,21 @@ if gender == 'Female':
     voice = female_voice
 else:
     voice = male_voice
-input_user = None
-messages = None
+
 if sumbit_btn:
     messages = st.container(height=300)
     input_user = st.chat_input(placeholder='Enter message for your partner...')
-    
-if input_user:
-    answer = gptClient.create_completion("gpt3", f"Answer on this question as beautiful nice kind {gender}: " + input_user)
-    print("Answer generated. Generating audio")
-    output_audio = get_speech(input_user, answer)
-    print("Audio generated. Generating video")
-    output_video = get_dreamtalk(img_prompt, output_audio)
-    print("Video Generating. Sending to user")
+    messages = st.container(height=300)
+
+    if input_user:
+        answer = gptClient.create_completion("gpt3", f"Answer on this question as beautiful nice kind {gender}: " + input_user)
+        print("Answer generated. Generating audio")
+        output_audio = get_speech(input_user, answer)
+        print("Audio generated. Generating video")
+        output_video = get_dreamtalk(img_prompt, output_audio)
+        print("Video Generating. Sending to user")
 
-    with messages.message("user"):
-        st.write(input_user)
-    with messages.message("Partner", avatar=img_prompt):
-        st.video(output_video)
+        with messages.message("user"):
+            st.write(input_user)
+        with messages.message("Partner", avatar=img_prompt):
+            st.video(output_video)