Update app.py

app.py

@@ -49,27 +49,37 @@ def fetch_user_data(handle):
 
 
 def fetch_and_save_data():
-    """Fetch all kinds of data for every user and save to a file."""
+    """Fetch all kinds of data for every user and save it to a file."""
     users = fetch_all_users()
     if not users:
         return "Failed to fetch users. Check logs for errors."
 
-    all_data = {}
+    if os.path.exists(DATA_FILE):
+        # Load existing data to resume
+        with open(DATA_FILE, "r", encoding="utf-8") as f:
+            all_data = json.load(f)
+    else:
+        all_data = {}
 
     for i, user in enumerate(users):
+        if user in all_data:
+            print(f"Skipping already fetched user: {user}")
+            continue
+
         try:
             user_data = fetch_user_data(user)
             if user_data:
                 all_data[user] = user_data
+
+            # Save progress to file after each user
+            with open(DATA_FILE, "w", encoding="utf-8") as f:
+                json.dump(all_data, f, indent=4)
+
             time.sleep(0.5)  # Respect API rate limit
         except Exception as e:
             print(f"Error fetching data for user {user}: {e}")
 
-    # Save data to JSON file
-    with open(DATA_FILE, "w", encoding="utf-8") as f:
-        json.dump(all_data, f, indent=4)
-
-    return f"Data fetching complete. Saved to {DATA_FILE}."
+    return "Data fetching complete."
 
 
 def serve_file():
@@ -88,15 +98,28 @@ def fetch_and_download():
     return message, file_path
 
 
+def get_file_for_download():
+    """Allow the user to download the current work anytime."""
+    file_path = serve_file()
+    if file_path:
+        return file_path
+    else:
+        return "No data file available."
+
+
 with gr.Blocks() as interface:
     gr.Markdown("# Codeforces User Data Fetcher")
-    gr.Markdown("This tool fetches data for all users from Codeforces and provides a downloadable file.")
-    
+    gr.Markdown("This tool fetches data for all users from Codeforces and provides a downloadable file. You can fetch data and download progress anytime.")
+
     with gr.Row():
         fetch_button = gr.Button("Fetch Data")
-        output_message = gr.Textbox(label="Status")
-        download_file = gr.File(label="Download Data", visible=False)
+        output_message = gr.Textbox(label="Status", lines=2)
+    
+    download_file_button = gr.Button("Download Current Data")
+    download_file_output = gr.File(label="Download Data", visible=False)
 
-    fetch_button.click(fetch_and_download, inputs=[], outputs=[output_message, download_file])
+    # Bind actions to buttons
+    fetch_button.click(fetch_and_download, inputs=[], outputs=[output_message, download_file_output])
+    download_file_button.click(get_file_for_download, inputs=[], outputs=download_file_output)
 
-interface.launch(share=True)
+interface.launch()