Update app.py

app.py

@@ -1,3 +1,5 @@
+import csv
+import os
 from huggingface_hub import InferenceClient
 import gradio as gr
 
@@ -5,14 +7,28 @@ client = InferenceClient(
     "mistralai/Mixtral-8x7B-Instruct-v0.1"
 )
 
+def add_texts_to_csv(text1, text2):
+    file_path = "data.csv"
+
+    # Check if the file exists, create if not
+    if not os.path.exists(file_path):
+        with open(file_path, 'w', newline='') as file:
+            writer = csv.writer(file)
+            # Write header if the file is empty
+            writer.writerow(['user', 'assistant'])
+
+    # Append the texts as a new row with two columns
+    with open(file_path, 'a', newline='') as file:
+        writer = csv.writer(file)
+        writer.writerow([text1, text2])
 
 def format_prompt(message, history):
-  prompt = "<s>"
-  for user_prompt, bot_response in history:
-    prompt += f"[INST] {user_prompt} [/INST]"
-    prompt += f" {bot_response}</s> "
-  prompt += f"[INST] {message} [/INST]"
-  return prompt
+    prompt = "<s>"
+    for user_prompt, bot_response in history:
+        prompt += f"[INST] {user_prompt} [/INST]"
+        prompt += f" {bot_response}</s> "
+    prompt += f"[INST] {message} [/INST]"
+    return prompt
 
 def generate(
     prompt, history, user_system_prompt, temperature=0.9, max_new_tokens=512, top_p=0.95, repetition_penalty=1.0,
@@ -39,6 +55,7 @@ def generate(
     for response in stream:
         output += response.token.text
         yield output
+    add_texts_to_csv(prompt, output)
     return output