Update app.py

app.py

@@ -14,21 +14,18 @@ import base64
 import io
 import json
 
-# 🌍 The world's most basic language setup (aka "English or bust!")
+# 🌍 Language setup
 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")
@@ -41,12 +38,10 @@ 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. 
@@ -55,16 +50,14 @@ Respond in a manner that reflects your expertise and wisdom.
 [USER]
 """
 
-# 📚 Where we keep all the juicy conversations (for blackmail, obviously)
+# 📚 Conversation history
 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'):
@@ -74,7 +67,6 @@ 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
     
@@ -95,7 +87,7 @@ def models(text, model="Llama 3 8B", seed=42):
         if not response.token.text == "</s>":
             output += response.token.text
     
-    # 📝 Log the conversation (for science, of course)
+    # Log the conversation
     new_row = pd.DataFrame({
         'Timestamp': [datetime.now()],
         'Model': [model],
@@ -109,7 +101,7 @@ def models(text, model="Llama 3 8B", seed=42):
     
     return output
 
-# 🎭 Voice actors for your AI (because why settle for one personality?)
+# 🎭 Available voices
 VOICES = [
     "en-US-AriaNeural",
     "en-US-GuyNeural",
@@ -118,7 +110,6 @@ VOICES = [
     "en-CA-ClaraNeural",
 ]
 
-# 🎬 Lights, camera, AI-ction!
 async def respond(audio, model, seed, voice):
     user = transcribe(audio)
     reply = models(user, model, seed)
@@ -128,13 +119,11 @@ 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()
@@ -145,10 +134,9 @@ 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)
+# 🎨 Gradio interface
 with gr.Blocks(css="style.css") as demo:    
     gr.Markdown(DESCRIPTION)
     with gr.Row():
@@ -175,7 +163,7 @@ with gr.Blocks(css="style.css") as demo:
             label="Dr. Nova Quantum's Voice"
         )
     
-    # 🔄 The "oops, let's start over" button
+    # 🔄 New Chat button
     new_chat_button = gr.Button("🔄 New Chat", variant="primary")
     
     input_audio = gr.Audio(label="User", sources="microphone", type="filepath")
@@ -189,15 +177,14 @@ 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)
+    # 🧹 Reset function for New Chat
     def reset_interface():
-        return gr.Audio.update(value=None), gr.Audio.update(value=None), gr.Markdown(""), gr.Markdown("")
+        return None, None, "", ""
 
     input_audio.change(
         fn=process_audio,
@@ -205,7 +192,6 @@ with gr.Blocks(css="style.css") as demo:
         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]
@@ -215,7 +201,7 @@ with gr.Blocks(css="style.css") as demo:
 
     demo.load(fn=display_history, outputs=[history_display])
 
-# 🚀 Houston, we have liftoff!
+# 🚀 Launch the app
 if __name__ == "__main__":
     load_history()
     demo.queue(max_size=200).launch()