Upload app.py with huggingface_hub

app.py

@@ -1155,6 +1155,30 @@ def load_model_manual():
     except Exception as e:
         logger.error(f"Manual model loading error: {str(e)}")
         return jsonify({'success': False, 'error': str(e)}), 500
+
+@app.route('/startup_check')
+def startup_check():
+    """Check startup status and trigger model loading if needed"""
+    try:
+        status = {
+            'model_loaded': chat_pipeline is not None,
+            'tokenizer_loaded': tokenizer is not None,
+            'langgraph_available': LANGGRAPH_AVAILABLE
+        }
+        
+        # If model not loaded, try to load it
+        if not chat_pipeline:
+            logger.info("Model not loaded, attempting to load...")
+            success = load_model()
+            status['model_load_attempted'] = True
+            status['model_load_success'] = success
+            status['model_loaded'] = chat_pipeline is not None
+        
+        return jsonify(status)
+        
+    except Exception as e:
+        logger.error(f"Startup check error: {str(e)}")
+        return jsonify({'error': str(e)}), 500
 @app.route('/health')
 def health():
     """Health check endpoint"""
@@ -1180,7 +1204,7 @@ if __name__ == '__main__':
     logger.info("Starting AEGIS Economics AI with LangGraph...")
     logger.info(f"LangGraph available: {LANGGRAPH_AVAILABLE}")
     
-    # Load model
+    # Load model immediately on startup
     logger.info("Loading model from Gaston895/Aegisecon1...")
     model_loaded = load_model()
     
@@ -1189,4 +1213,17 @@ if __name__ == '__main__':
     else:
         logger.error("❌ Model failed to load. Starting server anyway, but model endpoints will fail.")
     
-    app.run(host='0.0.0.0', port=7860, debug=False)
+    app.run(host='0.0.0.0', port=7860, debug=False)
+else:
+    # For production deployment (Gunicorn), load model when module is imported
+    logger.info("Production mode: Loading model during module import...")
+    logger.info(f"LangGraph available: {LANGGRAPH_AVAILABLE}")
+    
+    # Load model immediately
+    logger.info("Loading model from Gaston895/Aegisecon1...")
+    model_loaded = load_model()
+    
+    if model_loaded:
+        logger.info("✅ Model loaded successfully for production!")
+    else:
+        logger.error("❌ Model failed to load in production mode!")