Upload app.py with huggingface_hub

app.py

@@ -19,11 +19,20 @@ def check_daily_limit():
     today = datetime.now().date()
     limit_file = f"daily_limit_{today}.txt"
     
+    # For debugging, let's check what's in the file
     if os.path.exists(limit_file):
-        with open(limit_file, 'r') as f:
-            last_run = f.read().strip()
-        if last_run == str(today):
-            return False, "Daily H200 limit reached. Try again tomorrow!"
+        try:
+            with open(limit_file, 'r') as f:
+                last_run = f.read().strip()
+            print(f"Debug: Found limit file with content: '{last_run}' for date: {today}")
+            if last_run == str(today):
+                return False, f"Daily H200 limit reached. Try again tomorrow! (Last run: {last_run})"
+        except Exception as e:
+            print(f"Debug: Error reading limit file: {e}")
+            # If there's an error reading the file, let's allow training
+            return True, "Ready to train! (Limit file error, allowing training)"
+    else:
+        print(f"Debug: No limit file found for today: {today}")
     
     return True, "Ready to train!"
 
@@ -34,6 +43,18 @@ def mark_daily_usage():
     
     with open(limit_file, 'w') as f:
         f.write(str(today))
+    print(f"Debug: Marked daily usage for {today}")
+
+def reset_daily_limit():
+    """Reset the daily limit (for testing)."""
+    today = datetime.now().date()
+    limit_file = f"daily_limit_{today}.txt"
+    
+    if os.path.exists(limit_file):
+        os.remove(limit_file)
+        return f"✅ Daily limit reset for {today}"
+    else:
+        return f"ℹ️ No limit file found for {today}"
 
 def run_training():
     """Run the free H200 training."""
@@ -56,12 +77,19 @@ def run_training():
         print("Starting free H200 training...")
         start_time = time.time()
         
+        # Set environment variables for HF
+        env = os.environ.copy()
+        # HF Spaces automatically provides HF_TOKEN
+        if 'HF_TOKEN' not in env:
+            env['HF_TOKEN'] = os.environ.get('HF_TOKEN', '')
+        
         # Run training with timeout
         process = subprocess.Popen(
             ["python", TRAINING_SCRIPT],
             stdout=subprocess.PIPE,
             stderr=subprocess.STDOUT,
-            universal_newlines=True
+            universal_newlines=True,
+            env=env
         )
         
         output_lines = []
@@ -133,6 +161,7 @@ with gr.Blocks(title="Nano-Coder Free H200 Training") as demo:
         with gr.Column():
             gr.Markdown("### 🎯 Training Control")
             train_button = gr.Button("🚀 Start Free H200 Training", variant="primary")
+            reset_button = gr.Button("🔄 Reset Daily Limit", variant="secondary")
             status_text = gr.Textbox(label="Training Status", lines=10, interactive=False)
             
         with gr.Column():
@@ -160,6 +189,11 @@ with gr.Blocks(title="Nano-Coder Free H200 Training") as demo:
         outputs=status_text
     )
     
+    reset_button.click(
+        fn=reset_daily_limit,
+        outputs=status_text
+    )
+    
     model_status_button.click(
         fn=check_model_status,
         outputs=model_status_text