Update app.py

app.py

@@ -3,24 +3,25 @@ import gradio as gr
 
 from PIL import Image
 import os
+
+from share_btn import community_icon_html, loading_icon_html, share_js
+
 token = os.environ.get('HF_TOKEN')
 whisper_to_gpt = gr.Blocks.load(name="spaces/fffiloni/whisper-to-chatGPT")
 tts = gr.Interface.load(name="spaces/Flux9665/IMS-Toucan")
 talking_face = gr.Blocks.load(name="spaces/fffiloni/one-shot-talking-face", api_key=token)
 
-
-
 def infer(audio):
-    gpt_response = whisper_to_gpt(audio, "translate", fn_index=0)
+    whisper_to_gpt_response = whisper_to_gpt(audio, "translate", fn_index=0)
     #print(gpt_response)
-    audio_response = tts(gpt_response[1], "English Text", "English Accent", "English Speaker's Voice", fn_index=0)
+    audio_response = tts(whisper_to_gpt_response[1], "English Text", "English Accent", "English Speaker's Voice", fn_index=0)
     #image = Image.open(r"wise_woman_portrait.png")
     portrait_link = talking_face("wise_woman_portrait.png", audio_response, fn_index=0)
     
-    
     #portrait_response = requests.get(portrait_link, headers={'Authorization': 'Bearer ' + token})
     #print(portrait_response.text)
-    return portrait_link
+    
+    return whisper_to_gpt_response[0], portrait_link, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 
 title = """
     <div style="text-align: center; max-width: 500px; margin: 0 auto;">
@@ -80,15 +81,22 @@ with gr.Blocks(css=css) as demo:
         
         gr.HTML(title)
         
-        gpt_response = gr.Video(label="Talking Portrait response")
+        gpt_response = gr.Video(label="Talking Portrait response", elem_id="video_out")
              
     with gr.Column(elem_id="col-container-2"):
           
-        record_input = gr.Audio(source="microphone",type="filepath", label="Audio input", show_label=True,elem_id="record_btn")
+        record_input = gr.Audio(source="microphone",type="filepath", label="Audio input", show_label=True, elem_id="record_btn")
+        whisper_tr = gr.Textbox(label="whisper english translation", elem_id="text_inp")
 
         send_btn = gr.Button("Send my request !")
-   
-    send_btn.click(infer, inputs=[record_input], outputs=[gpt_response])
+    
+    with gr.Group(elem_id="share-btn-container"):
+            community_icon = gr.HTML(community_icon_html, visible=False)
+            loading_icon = gr.HTML(loading_icon_html, visible=False)
+            share_button = gr.Button("Share to community", elem_id="share-btn", visible=False)
+    
+    send_btn.click(infer, inputs=[record_input], outputs=[whisper_tr, gpt_response, share_button, community_icon, loading_icon])
+    share_button.click(None, [], [], _js=share_js)
 
 demo.queue(max_size=32, concurrency_count=20).launch(debug=True)