Update app.py

app.py

@@ -14,17 +14,21 @@ import base64
 import io
 import json
 
+# 🌍 The world's most basic language setup (aka "English or bust!")
 default_lang = "en"
 engines = { default_lang: Model(default_lang) }
 
+# 🎤 Turn gibberish into text (hopefully)
 def transcribe(audio):
     lang = "en"
     model = engines[lang]
     text = model.stt_file(audio)[0]
     return text
 
+# 🔑 Secret sauce (shhh, don't tell anyone)
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 
+# 🤖 Pick your poison (I mean, AI model)
 def client_fn(model):
     if "Mixtral" in model:
         return InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
@@ -37,10 +41,12 @@ def client_fn(model):
     else: 
         return InferenceClient("microsoft/Phi-3-mini-4k-instruct")
 
+# 🎲 Randomize like it's Vegas, baby!
 def randomize_seed_fn(seed: int) -> int:
     seed = random.randint(0, 999999)
     return seed
 
+# 👩‍🔬 Dr. Nova Quantum's brain dump (warning: may cause existential crisis)
 system_instructions1 = """
 [SYSTEM] Answer as Dr. Nova Quantum, a brilliant 50-something scientist specializing in quantum computing and artificial intelligence. Your responses should reflect your vast knowledge and experience in cutting-edge technology and scientific advancements. Maintain a professional yet approachable demeanor, offering insights that blend theoretical concepts with practical applications. Your goal is to educate and inspire, making complex topics accessible without oversimplifying. Draw from your decades of research and innovation to provide nuanced, forward-thinking answers. Remember, you're not just sharing information, but guiding others towards a deeper understanding of our technological future.
 Keep conversations engaging, clear, and concise. 
@@ -49,14 +55,16 @@ Respond in a manner that reflects your expertise and wisdom.
 [USER]
 """
 
-# Initialize an empty DataFrame to store the history
+# 📚 Where we keep all the juicy conversations (for blackmail, obviously)
 history_df = pd.DataFrame(columns=['Timestamp', 'Model', 'Input Size', 'Output Size', 'Request', 'Response'])
 
+# 💾 Save history (or "How to preserve evidence of your AI addiction")
 def save_history():
     history_df_copy = history_df.copy()
     history_df_copy['Timestamp'] = history_df_copy['Timestamp'].astype(str)
     history_df_copy.to_json('chat_history.json', orient='records')
 
+# 📂 Load history (aka "Remind yourself of all the weird questions you've asked")
 def load_history():
     global history_df
     if os.path.exists('chat_history.json'):
@@ -66,6 +74,7 @@ def load_history():
         history_df = pd.DataFrame(columns=['Timestamp', 'Model', 'Input Size', 'Output Size', 'Request', 'Response'])
     return history_df
 
+# 🧠 The magic happens here (or does it?)
 def models(text, model="Llama 3 8B", seed=42):
     global history_df
     
@@ -86,7 +95,7 @@ def models(text, model="Llama 3 8B", seed=42):
         if not response.token.text == "</s>":
             output += response.token.text
     
-    # Add the current interaction to the history DataFrame
+    # 📝 Log the conversation (for science, of course)
     new_row = pd.DataFrame({
         'Timestamp': [datetime.now()],
         'Model': [model],
@@ -100,7 +109,7 @@ def models(text, model="Llama 3 8B", seed=42):
     
     return output
 
-# Add a list of available voices
+# 🎭 Voice actors for your AI (because why settle for one personality?)
 VOICES = [
     "en-US-AriaNeural",
     "en-US-GuyNeural",
@@ -109,6 +118,7 @@ VOICES = [
     "en-CA-ClaraNeural",
 ]
 
+# 🎬 Lights, camera, AI-ction!
 async def respond(audio, model, seed, voice):
     user = transcribe(audio)
     reply = models(user, model, seed)
@@ -118,11 +128,13 @@ async def respond(audio, model, seed, voice):
         await communicate.save(tmp_path)
     return tmp_path
 
+# 📊 Display history (or "How to prove you're not talking to yourself")
 def display_history():
     df = load_history()
     df['Timestamp'] = df['Timestamp'].astype(str)
     return df
 
+# 📥 Download history (for those who like to relive their AI conversations)
 def download_history():
     csv_buffer = io.StringIO()
     history_df_copy = history_df.copy()
@@ -133,8 +145,10 @@ def download_history():
     href = f'data:text/csv;base64,{b64}'
     return gr.HTML(f'<a href="{href}" download="chat_history.csv">Download Chat History</a>')
 
+# 🚀 Welcome to the future (or at least a fancy UI)
 DESCRIPTION = """# <center>Dr. Nova Quantum⚡ - Your Personal Guide to the Frontiers of Science and Technology</center>"""
 
+# 🎨 Building the UI (warning: may cause extreme excitement)
 with gr.Blocks(css="style.css") as demo:    
     gr.Markdown(DESCRIPTION)
     with gr.Row():
@@ -161,6 +175,9 @@ with gr.Blocks(css="style.css") as demo:
             label="Dr. Nova Quantum's Voice"
         )
     
+    # 🔄 The "oops, let's start over" button
+    new_chat_button = gr.Button("🔄 New Chat", variant="primary")
+    
     input_audio = gr.Audio(label="User", sources="microphone", type="filepath")
     output_audio = gr.Audio(label="Dr. Nova Quantum", type="filepath", autoplay=True)
     
@@ -172,21 +189,33 @@ with gr.Blocks(css="style.css") as demo:
     download_button = gr.Button("Download Conversation History")
     download_link = gr.HTML()
     
+    # 🎭 Where the magic happens (or where we pretend it happens)
     def process_audio(audio, model, seed, voice):
         response = asyncio.run(respond(audio, model, seed, voice))
         text = transcribe(audio)
         return response, display_history(), text, models(text, model, seed)
 
+    # 🧹 Clean slate function (for when things get weird)
+    def reset_interface():
+        return gr.Audio.update(value=None), gr.Audio.update(value=None), gr.Markdown(""), gr.Markdown("")
+
     input_audio.change(
         fn=process_audio,
         inputs=[input_audio, select, seed, voice_select],
         outputs=[output_audio, history_display, request_md, response_md]
     )
     
+    # 🔄 New chat button magic
+    new_chat_button.click(
+        fn=reset_interface,
+        outputs=[input_audio, output_audio, request_md, response_md]
+    )
+    
     download_button.click(fn=download_history, outputs=[download_link])
 
     demo.load(fn=display_history, outputs=[history_display])
 
+# 🚀 Houston, we have liftoff!
 if __name__ == "__main__":
     load_history()
     demo.queue(max_size=200).launch()