Update app.py

app.py

@@ -23,7 +23,7 @@ def transcribe(audio):
     text = model.stt_file(audio)[0]
     return text
 
-HF_TOKEN = os.environ.get("HF_TOKEN", None)
+HF_TOKEN = os.environ.get("NOVA_TOKEN", None)
 
 def client_fn(model):
     if "Mixtral" in model:
@@ -52,7 +52,7 @@ Respond in a manner that reflects your expertise and wisdom.
 """
 
 # Initialize an empty DataFrame to store the history
-history_df = pd.DataFrame(columns=['Timestamp', 'Request', 'Response', 'Model'])
+history_df = pd.DataFrame(columns=['Timestamp', 'Model', 'Request', 'Response', 'Input Size', 'Output Size'])
 
 def save_history():
     history_df_copy = history_df.copy()
@@ -65,7 +65,7 @@ def load_history():
         history_df = pd.read_json('chat_history.json', orient='records')
         history_df['Timestamp'] = pd.to_datetime(history_df['Timestamp'])
     else:
-        history_df = pd.DataFrame(columns=['Timestamp', 'Request', 'Response', 'Model'])
+        history_df = pd.DataFrame(columns=['Timestamp', 'Model', 'Request', 'Response', 'Input Size', 'Output Size'])
     return history_df
 
 def models(text, model="Meta-Llama-3.1-8B-Instruct", seed=42):
@@ -91,9 +91,11 @@ def models(text, model="Meta-Llama-3.1-8B-Instruct", seed=42):
     # Add the current interaction to the history DataFrame
     new_row = pd.DataFrame({
         'Timestamp': [datetime.now()],
+        'Model': [model],
         'Request': [text],
         'Response': [output],
-        'Model': [model]
+        'Input Size': [len(text)],
+        'Output Size': [len(output)]
     })
     history_df = pd.concat([history_df, new_row], ignore_index=True)
     save_history()
@@ -155,8 +157,8 @@ with gr.Blocks(css="style.css") as demo:
     request_md = gr.Markdown(label="User Request")
     response_md = gr.Markdown(label="Dr. Nova Quantum Response")
     
-    # Update the DataFrame to display the history with the new model column
-    history_display = gr.DataFrame(label="Conversation History", headers=["Timestamp", "Request", "Response", "Model"])
+    # Update the DataFrame to display the history with the new order and additional columns
+    history_display = gr.DataFrame(label="Conversation History", headers=["Timestamp", "Model", "Request", "Response", "Input Size", "Output Size"])
     
     # Add a download button for the history
     download_button = gr.Button("Download Conversation History")