Update app.py

app.py

@@ -377,14 +377,23 @@ def generate(
     print(groupNrStart)
     print(playerNr)
 
+
     # Save chat history to .csv file on HuggingFace Hub 
     # Generate filename with bot id and session id
     filename = f"{groupNrStart}_{playerNr}_{PROLIFIC_PID}_{onPage}_{DATA_FILENAME}"
     data_file = os.path.join(DATA_DIRECTORY, filename)
-
+    
     # Generate timestamp
     timestamp = datetime.datetime.now()
     
+    # Check if the file already exists
+    if os.path.exists(data_file):
+        # If file exists, load existing data
+        existing_data = pd.read_csv(data_file)
+    else:
+        # If file doesn't exist, set existing_data to None
+        existing_data = None
+    
     # Create a DataFrame for the current conversation turn
     turn_data = {
         "turn_id": len(existing_data) + 1 if existing_data is not None else 1,
@@ -396,14 +405,10 @@ def generate(
     }
     turn_df = pd.DataFrame([turn_data])
     
-    # Check if the file already exists
-    if os.path.exists(data_file):
-        # If file exists, load existing data
-        existing_data = pd.read_csv(data_file)
-        # Append the new conversation turn to the existing data
+    # Check if existing_data is not None and concatenate the new conversation turn
+    if existing_data is not None:
         updated_data = pd.concat([existing_data, turn_df], ignore_index=True)
     else:
-        # If file doesn't exist, create new DataFrame with the current conversation turn
         updated_data = turn_df
     
     # Write the updated data to the CSV file