Update app.py

app.py

@@ -29,6 +29,11 @@ def get_file_size(filename):
 # Data generation and saving function
 def generate_and_save_data():
     global file_index, current_file, file_paths, combined_tokens
+    
+    # Create the initial file if it doesn't exist
+    if not os.path.isfile(current_file):
+        pd.DataFrame(columns=["prompt", "response"]).to_csv(current_file, index=False)
+
     while True:
         try:
             # Generate a prompt
@@ -94,11 +99,11 @@ def generate_and_save_data():
                 file_index += 1
                 current_file = os.path.join(DATA_DIRECTORY, f'data{file_index}.csv')
                 file_paths.append(current_file)
-
-            # Append data to the current file
-            mode = 'a' if os.path.isfile(current_file) else 'w'
-            header = not os.path.isfile(current_file)
-            data.to_csv(current_file, mode=mode, header=header, index=False)
+                # Create the new file with headers
+                data.to_csv(current_file, mode='w', header=True, index=False)
+            else:
+                # Append data to the current file
+                data.to_csv(current_file, mode='a', header=False, index=False)
 
             # Wait for the next update interval
             time.sleep(UPDATE_INTERVAL)