Update app.py

app.py

@@ -74,6 +74,7 @@ def process_audio(audio:tuple,state:AppState,image:Image):
         pause_detected = determine_pause(state.stream, state.sampling_rate, state)
         state.pause_detected = pause_detected
         if state.pause_detected and state.started_talking:
+            state.started_talking = False
             state.recording = False
             return state, gr.Audio(recording=False)
     return state, None
@@ -263,7 +264,7 @@ def update_model_choices(provider):
     }
     models = provider_models.get(provider, [])
     return gr.Dropdown(choices=models, value=models[0] if models else "")
-
+prompt_types = ["Long", "Short", "Medium", "OnlyObjects", "NoFigure", "Landscape", "Fantasy"]
 title = "Chat tiếng việt by tuphamkts"
 description = "Muốn vẽ nói: Chuyển sang chế độ vẽ. Muốn chat nói: Chuyển sang chế độ nói. Chế độ gõ: Tôi muốn dùng bàn phím, chế độ nói: Tôi muốn nói chuyện. Ghi chú: Chỉ dừng chương trình khi tôi đang nói (lịch sử chat sẽ bị xóa khi dừng chương trình)."
 examples = ["Chuyển sang chế độ vẽ","Chuyển sang chế độ nói"]
@@ -275,7 +276,6 @@ with gr.Blocks(title=title) as demo:
                 with gr.Row():
                     llm_provider = gr.Dropdown(choices=["Hugging Face", "SambaNova"], label="Nguồn model", value="Hugging Face")
                     model = gr.Dropdown(label="Chọn Model", choices=["Qwen/Qwen2.5-72B-Instruct","meta-llama/Meta-Llama-3.1-70B-Instruct","mistralai/Mixtral-8x7B-Instruct-v0.1","mistralai/Mistral-7B-Instruct-v0.3"], value="Qwen/Qwen2.5-72B-Instruct")
-                    prompt_types = ["Long", "Short", "Medium", "OnlyObjects", "NoFigure", "Landscape", "Fantasy"]
                     prompt_type = gr.Dropdown(choices=prompt_types, label="Phong cách", value="Medium", interactive=True)
                 input_prompt = gr.Textbox(label="Nhập nội dung muốn vẽ",value="Một cô gái", type="text"),
                 generate_prompt = gr.Button("Tạo Prompt", variant="stop")
@@ -298,16 +298,16 @@ with gr.Blocks(title=title) as demo:
     )
     stream = input_audio.stream(
         process_audio,
-        [input_audio, state,input_image],
+        [input_audio,state,input_image],
         [state,input_audio],
         stream_every=1,
         time_limit=30,
     )
 
     respond = input_audio.stop_recording(
-        fn=response_audio,
-        inputs=[state],
-        outputs=[state, output_audio],
+        response_audio,
+        [state],
+        [state, output_audio],
     )
     respond.then(lambda s: s.conversation, [state], [chatbot])
     respond.then(lambda s: s.image_out, [state], [output_image])
@@ -328,27 +328,27 @@ with gr.Blocks(title=title) as demo:
     cancel.click(
         lambda: (AppState(stopped=True, recording=False, started_talking = False), gr.Audio(recording=False), gr.update(interactive=False)), 
         None,[state, input_audio, cancel],
-        cancels=[respond, stream, startrecord]  # Thêm startrecord và stream vào cancels
+        cancels=[respond, stream, startrecord, restart]  # Thêm startrecord và stream vào cancels
     )
     
     sub = submit.click(
-        fn=response_text, 
-        inputs=[state, input_text[0], input_image, input_prompt[0]],
-        outputs=[state, output_audio]  
+        response_text, 
+        [state, input_text[0], input_image, input_prompt[0]],
+        [state, output_audio], 
     )
     sub.then(lambda s: s.conversation, [state], [chatbot])
     sub.then(lambda s: s.image_out, [state], [output_image])
     
     generator = generate_prompt.click(
-        fn=prompt_hugingface, 
-        inputs=[input_prompt[0],llm_provider,model,prompt_type],
-        outputs=[input_text[0]]  
+        prompt_hugingface, 
+        [input_prompt[0],llm_provider,model,prompt_type],
+        [input_text[0]]  
     )
     
     llm_provider.change(
         update_model_choices, 
-        inputs=[llm_provider], 
-        outputs=[model]
+        [llm_provider], 
+        [model]
     )
     gr.Examples(
         examples=examples,
@@ -356,4 +356,4 @@ with gr.Blocks(title=title) as demo:
     )
     
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860, share=False)
+    demo.launch()