Update app.py

app.py

@@ -10,8 +10,15 @@ from langchain_groq import ChatGroq
 import chardet
 import pandas as pd
 import plotly.graph_objs as go
+import shutil
 
-CSV_PATH = "/home/user/evaluations.csv"
+TMP_CSV_PATH = "/tmp/evaluations.csv"
+PERSISTENT_CSV_PATH = "/home/user/evaluations.csv"
+
+def ensure_directory_exists(file_path):
+    directory = os.path.dirname(file_path)
+    if not os.path.exists(directory):
+        os.makedirs(directory)
 
 def detect_encoding(file_path):
     with open(file_path, 'rb') as file:
@@ -106,11 +113,12 @@ def chat_with_models(file, model_a, model_b, api_key, history_a, history_b, ques
         return history_a + [(error_message, None)], history_b + [(error_message, None)], ""
 
 def load_or_create_df():
-    if os.path.exists(CSV_PATH):
-        return pd.read_csv(CSV_PATH)
+    if os.path.exists(PERSISTENT_CSV_PATH):
+        return pd.read_csv(PERSISTENT_CSV_PATH)
     else:
         df = pd.DataFrame(columns=['Model A', 'Model B', 'Evaluation', 'Count'])
-        df.to_csv(CSV_PATH, index=False)
+        ensure_directory_exists(PERSISTENT_CSV_PATH)
+        df.to_csv(PERSISTENT_CSV_PATH, index=False)
         return df
 
 def record_evaluation(df, model_a, model_b, evaluation):
@@ -121,9 +129,15 @@ def record_evaluation(df, model_a, model_b, evaluation):
         'Count': [1]
     })
     updated_df = pd.concat([df, new_row], ignore_index=True)
-    updated_df.to_csv(CSV_PATH, index=False)
+    ensure_directory_exists(TMP_CSV_PATH)
+    updated_df.to_csv(TMP_CSV_PATH, index=False)
+    copy_to_persistent_storage()
     return updated_df
 
+def copy_to_persistent_storage():
+    ensure_directory_exists(PERSISTENT_CSV_PATH)
+    shutil.copy2(TMP_CSV_PATH, PERSISTENT_CSV_PATH)
+
 def update_statistics(df):
     stats = df.groupby(['Model A', 'Model B', 'Evaluation'])['Count'].sum().reset_index()
     
@@ -145,7 +159,8 @@ def evaluate(df, model_a, model_b, evaluation):
 models = ["llama3-70b-8192", "mixtral-8x7b-32768", "llama3-8b-8192", "gemma-7b-it"]
 
 def create_demo():
-    print(f"CSV file is located at: {CSV_PATH}")
+    print(f"Temporary CSV file is located at: {TMP_CSV_PATH}")
+    print(f"Persistent CSV file is located at: {PERSISTENT_CSV_PATH}")
 
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("# Choose two models to compare")
@@ -196,7 +211,7 @@ def create_demo():
     return demo
 
 if __name__ == "__main__":
-    print(f"CSV file is located at: {CSV_PATH}")
+    print(f"CSV file is located at: {PERSISTENT_CSV_PATH}")
     demo = create_demo()
     demo.launch(share=True)
 else: