Update app.py

app.py

@@ -1,3 +1,8 @@
+
+
+
+
+
 """
 FastAPI webhook server for Hugging Face Spaces
 Handles training requests from Vercel and sends results back via callback
@@ -37,7 +42,6 @@ app.add_middleware(
     allow_methods=["*"],
     allow_headers=["*"],
 )
-
 # Training state
 training_in_progress = False
 training_result = None
@@ -84,19 +88,38 @@ def run_training_async(training_data: str, callback_url: str, callback_auth_toke
         
         # Send callback to Vercel
         logger.info(f"Sending callback to {callback_url}...")
-        response = requests.post(
-            callback_url,
-            json=callback_payload,
-            timeout=60,  # Reduced timeout since we're not generating predictions
-            headers={'Content-Type': 'application/json'}
-        )
+        logger.info(f"Callback payload stats: accuracy={stats['accuracy']:.4f}, loss={stats['loss']:.4f}, f1_scores={list(stats['f1_scores'].keys())}")
         
-        if response.status_code == 200:
-            logger.info("Callback sent successfully")
-            training_result = {'success': True, 'message': 'Training completed'}
-        else:
-            logger.error(f"Callback failed with status {response.status_code}: {response.text}")
-            training_error = f"Callback failed: {response.status_code}"
+        try:
+            response = requests.post(
+                callback_url,
+                json=callback_payload,
+                timeout=60,  # Reduced timeout since we're not generating predictions
+                headers={'Content-Type': 'application/json'}
+            )
+            
+            logger.info(f"Callback response status: {response.status_code}")
+            logger.info(f"Callback response headers: {dict(response.headers)}")
+            logger.info(f"Callback response body: {response.text}")
+            
+            if response.status_code == 200:
+                try:
+                    response_json = response.json()
+                    logger.info(f"Callback response JSON: {response_json}")
+                    if response_json.get('success'):
+                        logger.info("Callback sent successfully and processed by Vercel")
+                    else:
+                        logger.warning(f"Callback returned 200 but success=false: {response_json}")
+                except Exception as e:
+                    logger.warning(f"Callback returned 200 but couldn't parse JSON: {e}, body: {response.text}")
+                training_result = {'success': True, 'message': 'Training completed'}
+            else:
+                logger.error(f"Callback failed with status {response.status_code}: {response.text}")
+                training_error = f"Callback failed: {response.status_code}"
+                training_result = {'success': False, 'error': training_error}
+        except requests.exceptions.RequestException as e:
+            logger.error(f"Callback request exception: {str(e)}", exc_info=True)
+            training_error = f"Callback request failed: {str(e)}"
             training_result = {'success': False, 'error': training_error}
         
     except Exception as e: