Update app.py

app.py

@@ -37,6 +37,52 @@ print(f"Loaded {len(train_texts)} examples from {data_file}")
 # Model save directory
 model_save_dir = "./results/model"
 
+@app.route('/')
+def home():
+    """Root endpoint to show API status and usage"""
+    return jsonify({
+        'status': 'SEAL Framework API is running',
+        'version': '1.0.0',
+        'model': model_name,
+        'device': str(device),
+        'training_examples': len(train_texts),
+        'endpoints': {
+            '/': 'GET - API status and information',
+            '/adapt': 'POST - Adaptive model training and response',
+            '/health': 'GET - Health check'
+        },
+        'usage': {
+            'adapt_endpoint': {
+                'method': 'POST',
+                'content_type': 'application/json',
+                'body': {'text': 'Your input text here'},
+                'example': 'curl -X POST -H "Content-Type: application/json" -d \'{"text":"Hello world"}\' /adapt'
+            }
+        }
+    })
+
+@app.route('/health')
+def health():
+    """Health check endpoint"""
+    try:
+        # Simple model test
+        test_input = "Health check"
+        inputs = tokenizer(test_input, return_tensors="pt", truncation=True, max_length=32).to(device)
+        with torch.no_grad():
+            outputs = model.generate(**inputs, max_length=40, num_return_sequences=1, do_sample=False)
+        
+        return jsonify({
+            'status': 'healthy',
+            'model_loaded': True,
+            'device': str(device),
+            'training_examples': len(train_texts)
+        })
+    except Exception as e:
+        return jsonify({
+            'status': 'unhealthy',
+            'error': str(e)
+        }), 500
+
 @app.route('/adapt', methods=['POST'])
 def adapt_model():
     try:
@@ -104,11 +150,33 @@ def adapt_model():
         return jsonify({
             'input': user_input,
             'self_edit': self_edit,
-            'response': response
+            'response': response,
+            'training_examples': len(train_texts),
+            'status': 'Model adapted successfully'
         })
 
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
+@app.errorhandler(404)
+def not_found(error):
+    """Custom 404 handler"""
+    return jsonify({
+        'error': 'Endpoint not found',
+        'available_endpoints': {
+            '/': 'GET - API information',
+            '/health': 'GET - Health check',
+            '/adapt': 'POST - Adaptive model training'
+        }
+    }), 404
+
+@app.errorhandler(500)
+def internal_error(error):
+    """Custom 500 handler"""
+    return jsonify({
+        'error': 'Internal server error',
+        'message': 'Please check the server logs for more details'
+    }), 500
+
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+    app.run(host='0.0.0.0', port=7860, debug=False)