Update app.py

app.py

@@ -11,7 +11,40 @@ import random
 import pandas as pd
 from datetime import datetime
 
-# ... (previous code remains the same)
+default_lang = "en"
+engines = { default_lang: Model(default_lang) }
+
+def transcribe(audio):
+    lang = "en"
+    model = engines[lang]
+    text = model.stt_file(audio)[0]
+    return text
+
+HF_TOKEN = os.environ.get("HF_TOKEN", None)
+
+def client_fn(model):
+    if "Mixtral" in model:
+        return InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
+    elif "Llama" in model:
+        return InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct")
+    elif "Mistral" in model:
+        return InferenceClient("mistralai/Mistral-7B-Instruct-v0.2")
+    elif "Phi" in model:
+        return InferenceClient("microsoft/Phi-3-mini-4k-instruct")
+    else: 
+        return InferenceClient("microsoft/Phi-3-mini-4k-instruct")
+
+def randomize_seed_fn(seed: int) -> int:
+    seed = random.randint(0, 999999)
+    return seed
+
+system_instructions1 = """
+[SYSTEM] Answer as Real Jarvis JARVIS, Made by 'Tony Stark.' 
+Keep conversation friendly, short, clear, and concise. 
+Avoid unnecessary introductions and answer the user's questions directly. 
+Respond in a normal, conversational manner while being friendly and helpful.
+[USER]
+"""
 
 # Initialize an empty DataFrame to store the history
 history_df = pd.DataFrame(columns=['Timestamp', 'Request', 'Response'])
@@ -61,7 +94,57 @@ def display_history():
 def download_history():
     return history_df.to_csv(index=False)
 
-# ... (rest of the code remains the same)
+DESCRIPTION = """ # <center><b>JARVIS⚡</b></center>
+        ### <center>A personal Assistant of Tony Stark for YOU
+        ### <center>Voice Chat with your personal Assistant</center>
+        """
+
+with gr.Blocks(css="style.css") as demo:    
+    gr.Markdown(DESCRIPTION)
+    with gr.Row():
+        select = gr.Dropdown([
+            'Mixtral 8x7B',
+            'Llama 3 8B',
+            'Mistral 7B v0.3',
+            'Phi 3 mini',
+        ],
+        value="Mistral 7B v0.3",
+        label="Model"
+        )
+        seed = gr.Slider(
+            label="Seed",
+            minimum=0,
+            maximum=999999,
+            step=1,
+            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")
+    
+    # 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")
+    
+    gr.Interface(
+        batch=True,
+        max_batch_size=10, 
+        fn=respond, 
+        inputs=[input, select, seed],
+        outputs=[output], 
+        live=True
+    )
+    
+    # Update the history display after each interaction
+    output.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")])
 
 if __name__ == "__main__":
     demo.queue(max_size=200).launch(share=True)  # Added share=True for public link