Update app.py

app.py

@@ -10,6 +10,8 @@ import torch
 import random
 import pandas as pd
 from datetime import datetime
+import base64
+import io
 
 default_lang = "en"
 engines = { default_lang: Model(default_lang) }
@@ -86,13 +88,18 @@ async def respond(audio, model, seed):
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
         tmp_path = tmp_file.name
         await communicate.save(tmp_path)
-    yield tmp_path
+    return tmp_path
 
 def display_history():
     return history_df
 
 def download_history():
-    return history_df.to_csv(index=False)
+    csv_buffer = io.StringIO()
+    history_df.to_csv(csv_buffer, index=False)
+    csv_string = csv_buffer.getvalue()
+    b64 = base64.b64encode(csv_string.encode()).decode()
+    href = f'data:text/csv;base64,{b64}'
+    return gr.HTML(f'<a href="{href}" download="chat_history.csv">Download Chat History</a>')
 
 DESCRIPTION = """ # <center><b>JARVIS⚡</b></center>
         ### <center>A personal Assistant of Tony Stark for YOU
@@ -119,32 +126,32 @@ with gr.Blocks(css="style.css") as demo:
             value=0,
             visible=False
         )
-        input = gr.Audio(label="User", sources="microphone", type="filepath", waveform_options=False)
-        output = gr.Audio(label="AI", type="filepath",
-                          interactive=False,
-                          autoplay=True,
-                          elem_classes="audio")
+    
+    input_audio = gr.Audio(label="User", sources="microphone", type="filepath")
+    output_audio = gr.Audio(label="AI", type="filepath", autoplay=True)
     
     # Add a DataFrame to display the history
     history_display = gr.DataFrame(label="Query History")
     
     # Add a download button for the history
     download_button = gr.Button("Download History")
+    download_link = gr.HTML()
     
-    gr.Interface(
-        batch=True,
-        max_batch_size=10, 
-        fn=respond, 
-        inputs=[input, select, seed],
-        outputs=[output], 
-        live=True
+    def process_audio(audio, model, seed):
+        response = asyncio.run(respond(audio, model, seed))
+        return response
+
+    input_audio.change(
+        fn=process_audio,
+        inputs=[input_audio, select, seed],
+        outputs=[output_audio]
     )
     
     # Update the history display after each interaction
-    output.change(fn=display_history, outputs=[history_display])
+    output_audio.change(fn=display_history, outputs=[history_display])
     
     # Connect the download button to the download function
-    download_button.click(fn=download_history, outputs=[gr.File(label="Download CSV")])
+    download_button.click(fn=download_history, outputs=[download_link])
 
 if __name__ == "__main__":
-    demo.queue(max_size=200).launch(share=True)  # Added share=True for public link
+    demo.queue(max_size=200).launch(share=True)