Update app.py

app.py

@@ -42,7 +42,7 @@ def generate_and_save_data():
                 messages=[
                     {
                         "role": "user",
-                        "content": "give me a single prompt to prompt an ai model, simulating what users could want from you. ensure that it is diverse and high quality. for each, choose a random writing style (though it has to be a common one), random length and random clarity of the prompt. ensure that I is a single prompt, and just the prompt itself, nothing else. eg, don't close the prompt in quotation marks or say Here is a single prompt that meets your requirements or anything similar to that"
+                        "content": "give me a single prompt to prompt an ai model, simulating what users could want from you. ensure that it is diverse and high quality. for each, choose a random writing style (though it has to be a common one), random length and random clarity of the prompt. ensure that it is a single prompt, and just the prompt itself, nothing else. eg, don't close the prompt in quotation marks or say Here is a single prompt that meets your requirements or anything similar to that"
                     }
                 ],
                 temperature=1,
@@ -102,12 +102,10 @@ def generate_and_save_data():
                 # Create the new file with headers
                 with open(current_file, 'w') as f:
                     data.to_csv(f, header=True, index=False)
-                    f.flush()
             else:
                 # Append data to the current file
                 with open(current_file, 'a') as f:
                     data.to_csv(f, header=False, index=False)
-                    f.flush()
 
             # Wait for the next update interval
             time.sleep(UPDATE_INTERVAL)
@@ -128,6 +126,12 @@ def update_file_list():
 def update_token_count():
     return combined_tokens
 
+# Display file content
+def display_file_content(selected_file):
+    if selected_file:
+        return pd.read_csv(selected_file)
+    return pd.DataFrame()
+
 # Start the data generation in a separate thread
 thread = threading.Thread(target=generate_and_save_data)
 thread.daemon = True
@@ -138,7 +142,8 @@ with gr.Blocks() as app:
     gr.Markdown("## AI Prompt and Response Generator")
     gr.Markdown("This app continuously generates AI prompts and responses, and writes them to CSV files.")
     
-    file_selector = gr.Dropdown(label="Select a data file to download", choices=get_available_files())
+    file_selector = gr.Dropdown(label="Select a data file to view and download", choices=get_available_files())
+    file_viewer = gr.DataFrame(label="CSV File Content")
     download_button = gr.File(label="Download Selected File")
     
     def download_file(selected_file):
@@ -146,6 +151,7 @@ with gr.Blocks() as app:
 
     refresh_button = gr.Button("Refresh File List")
     refresh_button.click(update_file_list, outputs=file_selector)
+    file_selector.change(display_file_content, inputs=file_selector, outputs=file_viewer)
     file_selector.change(download_file, inputs=file_selector, outputs=download_button)
     
     token_display = gr.Textbox(label="Combined Tokens", value=str(update_token_count()), interactive=False)