Update app.py

app.py

@@ -18,7 +18,7 @@ import threading
 
 # --- Login to Hugging Face using secret ---
 # Make sure HF_TOKEN is set in your Hugging Face Space > Settings > Repository secrets
-hf_token = os.environ.get("hugface")
+hf_token = os.environ.get("hugface") # Using "HF_TOKEN" is the standard on Spaces
 if not hf_token:
     raise ValueError("HF_TOKEN not found. Please set it in Hugging Face Space repository secrets.")
 login(token=hf_token)
@@ -221,4 +221,52 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Msaidizi wa Kiswahili") as demo:
                     gr.Markdown("### Utengenezaji wa Sauti (Speech Synthesis)")
                     tool_t2s_text_in = gr.Textbox(label="Maandishi ya Kuingiza (Input Text)", placeholder="Andika Kiswahili hapa...")
                     tool_t2s_audio_out = gr.Audio(type="filepath", label="Sauti Iliyotengenezwa (Synthesized Audio)", autoplay=False)
-                    tool_t2s_btn = gr.Button("Tengeneza Sauti
+                    tool_t2s_btn = gr.Button("Tengeneza Sauti (Synthesize)")
+
+    s2s_submit_btn.click(
+        fn=s2s_pipeline,
+        inputs=[s2s_audio_in, s2s_chatbot],
+        outputs=[s2s_chatbot, s2s_audio_out, s2s_text_out],
+        queue=True
+    ).then(
+        fn=lambda: gr.Audio(value=None),
+        inputs=None,
+        outputs=s2s_audio_in
+    )
+
+    t2t_submit_btn.click(
+        fn=t2t_pipeline,
+        inputs=[t2t_text_in, t2t_chatbot],
+        outputs=[t2t_chatbot],
+        queue=True
+    ).then(
+        fn=clear_textbox,
+        inputs=None,
+        outputs=t2t_text_in
+    )
+    
+    t2t_text_in.submit(
+        fn=t2t_pipeline,
+        inputs=[t2t_text_in, t2t_chatbot],
+        outputs=[t2t_chatbot],
+        queue=True
+    ).then(
+        fn=clear_textbox,
+        inputs=None,
+        outputs=t2t_text_in
+    )
+
+    tool_s2t_btn.click(
+        fn=assistant.transcribe_audio,
+        inputs=tool_s2t_audio_in,
+        outputs=tool_s2t_text_out,
+        queue=True
+    )
+    tool_t2s_btn.click(
+        fn=assistant.generate_speech,
+        inputs=tool_t2s_text_in,
+        outputs=tool_t2s_audio_out,
+        queue=True
+    )
+
+demo.queue().launch(debug=True)