Update app.py

app.py

@@ -6,17 +6,13 @@ from gtts import gTTS
 from transformers import pipeline
 from huggingface_hub import InferenceClient
 
-
 ASR_MODEL_NAME = "openai/whisper-small"
 LLM_MODEL_NAME = "mistralai/Mistral-7B-Instruct-v0.2"
 
-
-system_prompt = """"<s>[INST] You are Friday, a helpful and conversational AI assistant and You respond with one to two sentences. [/INST] Hello there! I'm friday how can I help you?</s>"""
+system_prompt = """"<s>[INST] You are Friday, a helpful and conversational AI assistant, and you respond with one to two sentences. [/INST] Hello there! I'm Friday, how can I help you?</s>"""
 
 instruct_history = system_prompt + """"""
 
-formatted_history = """"""
-
 client = InferenceClient(LLM_MODEL_NAME)
 
 device = 0 if torch.cuda.is_available() else "cpu"
@@ -27,7 +23,6 @@ pipe = pipeline(
     device=device,
 )
 
-
 def generate(instruct_history, temperature=0.1, max_new_tokens=128, top_p=0.95, repetition_penalty=1.0):
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -48,49 +43,48 @@ def generate(instruct_history, temperature=0.1, max_new_tokens=128, top_p=0.95,
 
     return output
 
-
 @spaces.GPU(duration=60)
-def transcribe(audio, instruct_history=instruct_history, formatted_history=formatted_history):
+def transcribe(audio, instruct_history=instruct_history):
     sr, y = audio
     y = y.astype(np.float32)
     y /= np.max(np.abs(y))
 
+    # Transcribe user audio
     transcribed_user_audio = pipe({"sampling_rate": sr, "raw": y})["text"]
 
-    formatted_history += f"""Human: {transcribed_user_audio}\n\n"""
-
+    # Append user input to history
+    formatted_history = f"""Human: {transcribed_user_audio}\n\n"""
     instruct_history += f"""<s>[INST] {transcribed_user_audio} [/INST] """
 
+    # Generate LLM response
     llm_response = generate(instruct_history)
 
+    # Append AI response to history
     instruct_history += f""" {llm_response}</s>"""
-
     formatted_history += f"""Friday: {llm_response}\n\n"""
 
+    # Convert AI response to audio
     audio_response = gTTS(llm_response)
     audio_response.save("response.mp3")
 
-    print(instruct_history)
-
-    return "response.mp3", formatted_history
+    # Display the full conversation history
+    full_history = formatted_history
 
+    return "response.mp3", full_history
 
 with gr.Blocks() as demo:
-    gr.HTML("<center><h1>Friday: AI Virtual Assistant<h1><center>")
+    gr.HTML("<center><h1>Friday: AI Virtual Assistant</h1><center>")
 
     with gr.Row():
-        audio_input = gr.Audio(label="Human", sources="microphone")
-        output_audio = gr.Audio(label="Friday", type="filepath",
-                                interactive=False,
-                                autoplay=True,
-                                elem_classes="audio")
+        audio_input = gr.Audio(label="Human", source="microphone")
+        output_audio = gr.Audio(label="Friday", type="filepath", interactive=False, autoplay=True, elem_classes="audio")
 
     transcribe_btn = gr.Button("Transcribe")
 
-    transcription_box = gr.Textbox(label="Transcription")
+    # Textbox to display the full conversation history
+    transcription_box = gr.Textbox(label="Transcription", lines=10, placeholder="Conversation History...")
 
-    transcribe_btn.click(fn=transcribe, inputs=[audio_input],
-                         outputs=[output_audio, transcription_box])
+    transcribe_btn.click(fn=transcribe, inputs=[audio_input], outputs=[output_audio, transcription_box])
 
 if __name__ == "__main__":
     demo.queue()