Update app.py

app.py

@@ -10,15 +10,13 @@ from langchain_groq import ChatGroq
 import chardet
 import pandas as pd
 import plotly.graph_objs as go
-import shutil
 
-TMP_CSV_PATH = "/tmp/evaluations.csv"
-PERSISTENT_CSV_PATH = "./evaluations.csv"
+try:
+    SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
+except NameError:
+    SCRIPT_DIR = os.getcwd()
 
-def ensure_directory_exists(file_path):
-    directory = os.path.dirname(file_path)
-    if not os.path.exists(directory):
-        os.makedirs(directory)
+CSV_PATH = os.path.join(SCRIPT_DIR, "evaluations.csv")
 
 def detect_encoding(file_path):
     with open(file_path, 'rb') as file:
@@ -113,13 +111,10 @@ 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(PERSISTENT_CSV_PATH):
-        return pd.read_csv(PERSISTENT_CSV_PATH)
+    if os.path.exists(CSV_PATH):
+        return pd.read_csv(CSV_PATH)
     else:
-        df = pd.DataFrame(columns=['Model A', 'Model B', 'Evaluation', 'Count'])
-        ensure_directory_exists(PERSISTENT_CSV_PATH)
-        df.to_csv(PERSISTENT_CSV_PATH, index=False)
-        return df
+        return pd.DataFrame(columns=['Model A', 'Model B', 'Evaluation', 'Count'])
 
 def record_evaluation(df, model_a, model_b, evaluation):
     new_row = pd.DataFrame({
@@ -129,15 +124,9 @@ def record_evaluation(df, model_a, model_b, evaluation):
         'Count': [1]
     })
     updated_df = pd.concat([df, new_row], ignore_index=True)
-    ensure_directory_exists(TMP_CSV_PATH)
-    updated_df.to_csv(TMP_CSV_PATH, index=False)
-    copy_to_persistent_storage()
+    updated_df.to_csv(CSV_PATH, index=False)
     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()
     
@@ -159,9 +148,6 @@ 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"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")
         
@@ -211,7 +197,7 @@ def create_demo():
     return demo
 
 if __name__ == "__main__":
-    print(f"CSV file is located at: {PERSISTENT_CSV_PATH}")
+    print(f"CSV file is located at: {os.path.abspath(CSV_PATH)}")
     demo = create_demo()
     demo.launch(share=True)
 else: