Upload folder using huggingface_hub

main.py

@@ -25,11 +25,6 @@ from huggingface_hub import hf_hub_download
 # Import your orchestration pipeline
 from orchestration import WBCClassificationPipeline
 
-# Configure logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-# Create FastAPI app instance
 app = FastAPI(
     title="WBC Classification API",
     description="AI-Powered White Blood Cell Classification System",
@@ -37,6 +32,11 @@ app = FastAPI(
     docs_url="/docs",
     redoc_url="/redoc"
 )
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+
 
 # Add CORS middleware
 app.add_middleware(
@@ -47,8 +47,32 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Configuration - FIXED PATHS
-MODEL_PATH = "./Models/at_batch_size=32.D24E.keras"  # Fixed path
+# Configuration - DYNAMIC PATHS FOR HUGGING FACE SPACES
+def get_model_path():
+    """Get model path dynamically based on environment"""
+    # For local development
+    local_path = "./Models/at_batch_size=32.D24E.keras"
+    if os.path.exists(local_path):
+        return local_path
+    
+    # For Hugging Face Spaces - model should be in root
+    hf_path = "at_batch_size=32.D24E.keras"
+    if os.path.exists(hf_path):
+        return hf_path
+    
+    # Alternative path in Models folder
+    models_path = "Models/at_batch_size=32.D24E.keras"
+    if os.path.exists(models_path):
+        return models_path
+    
+    # Last resort - current directory
+    current_dir_path = os.path.join(os.getcwd(), "at_batch_size=32.D24E.keras")
+    if os.path.exists(current_dir_path):
+        return current_dir_path
+    
+    return local_path  # Return default for error handling
+
+MODEL_PATH = get_model_path()
 STATIC_DIR = "static"
 UPLOADS_DIR = "uploads"
 MAX_FILE_SIZE = 10 * 1024 * 1024  # 10MB
@@ -71,13 +95,25 @@ async def startup_event():
     global pipeline
     
     logger.info("Starting WBC Classification API...")
+    logger.info(f"Current working directory: {os.getcwd()}")
     logger.info(f"Model path: {MODEL_PATH}")
+    logger.info(f"Files in current directory: {os.listdir('.')}")
+    
+    # List files in Models directory if it exists
+    if os.path.exists("Models"):
+        logger.info(f"Files in Models directory: {os.listdir('Models')}")
     
     try:
         # Check if model file exists
         if not os.path.exists(MODEL_PATH):
             logger.error(f"Model file not found at: {MODEL_PATH}")
-            logger.error("Please ensure your model file is in the Models folder")
+            logger.error("Please ensure your model file is accessible")
+            
+            # Try to find the model file
+            for root, dirs, files in os.walk("."):
+                for file in files:
+                    if file.endswith(".keras"):
+                        logger.info(f"Found keras file: {os.path.join(root, file)}")
             return
         
         # Initialize pipeline
@@ -101,21 +137,109 @@ async def startup_event():
 async def read_root():
     """Serve the main HTML page"""
     try:
-        html_path = os.path.join(STATIC_DIR, "index.html")
-        if os.path.exists(html_path):
-            with open(html_path, 'r', encoding='utf-8') as f:
-                html_content = f.read()
+        # Try multiple possible locations for the HTML file
+        possible_paths = [
+            os.path.join(STATIC_DIR, "index.html"),
+            "index.html",
+            os.path.join(".", "index.html")
+        ]
+        
+        html_content = None
+        for html_path in possible_paths:
+            if os.path.exists(html_path):
+                with open(html_path, 'r', encoding='utf-8') as f:
+                    html_content = f.read()
+                logger.info(f"Serving HTML from: {html_path}")
+                break
+        
+        if html_content:
             return HTMLResponse(content=html_content)
         else:
+            # Return a basic HTML page if index.html is not found
             return HTMLResponse(
                 content="""
-                <h1>🩸 WBC Classifier API</h1>
-                <p>Frontend HTML file not found in static folder.</p>
-                <p>Please move your index.html to the 'static' folder.</p>
-                <p><a href="/docs">View API Documentation</a></p>
+                <!DOCTYPE html>
+                <html>
+                <head>
+                    <title>WBC Classification API</title>
+                    <style>
+                        body { font-family: Arial, sans-serif; margin: 40px; }
+                        .container { max-width: 800px; margin: 0 auto; }
+                        .upload-area { border: 2px dashed #ccc; padding: 40px; text-align: center; margin: 20px 0; }
+                        .btn { background: #007bff; color: white; padding: 10px 20px; border: none; cursor: pointer; }
+                        .results { margin-top: 20px; padding: 20px; background: #f8f9fa; }
+                    </style>
+                </head>
+                <body>
+                    <div class="container">
+                        <h1>🩸 WBC Classification API</h1>
+                        <p>Upload white blood cell images for classification</p>
+                        
+                        <div class="upload-area">
+                            <input type="file" id="fileInput" accept="image/*" style="display: none;">
+                            <button class="btn" onclick="document.getElementById('fileInput').click()">
+                                Select Image
+                            </button>
+                            <p>Supported formats: JPG, PNG, BMP, TIFF, GIF</p>
+                        </div>
+                        
+                        <div id="results" class="results" style="display: none;"></div>
+                        
+                        <div>
+                            <h3>API Endpoints:</h3>
+                            <ul>
+                                <li><a href="/docs">API Documentation</a></li>
+                                <li><a href="/api/health">Health Check</a></li>
+                                <li><a href="/api/model_info">Model Information</a></li>
+                            </ul>
+                        </div>
+                    </div>
+                    
+                    <script>
+                        document.getElementById('fileInput').addEventListener('change', function(event) {
+                            const file = event.target.files[0];
+                            if (file) {
+                                const formData = new FormData();
+                                formData.append('file', file);
+                                
+                                fetch('/api/predict', {
+                                    method: 'POST',
+                                    body: formData
+                                })
+                                .then(response => response.json())
+                                .then(data => {
+                                    const resultsDiv = document.getElementById('results');
+                                    if (data.success) {
+                                        resultsDiv.innerHTML = `
+                                            <h3>Classification Results:</h3>
+                                            <p><strong>Predicted Class:</strong> ${data.results.predicted_class}</p>
+                                            <p><strong>Confidence:</strong> ${(data.results.confidence * 100).toFixed(2)}%</p>
+                                            <h4>All Probabilities:</h4>
+                                            <ul>
+                                                ${Object.entries(data.results.all_probabilities).map(([className, prob]) => 
+                                                    `<li>${className}: ${(prob * 100).toFixed(2)}%</li>`
+                                                ).join('')}
+                                            </ul>
+                                        `;
+                                    } else {
+                                        resultsDiv.innerHTML = `<p style="color: red;">Error: ${data.detail || 'Unknown error'}</p>`;
+                                    }
+                                    resultsDiv.style.display = 'block';
+                                })
+                                .catch(error => {
+                                    const resultsDiv = document.getElementById('results');
+                                    resultsDiv.innerHTML = `<p style="color: red;">Error: ${error.message}</p>`;
+                                    resultsDiv.style.display = 'block';
+                                });
+                            }
+                        });
+                    </script>
+                </body>
+                </html>
                 """,
                 status_code=200
             )
+            
     except Exception as e:
         logger.error(f"Error serving root page: {str(e)}")
         return HTMLResponse(
@@ -135,7 +259,9 @@ async def health_check():
             "status": "healthy" if model_loaded else "unhealthy",
             "model_loaded": model_loaded,
             "model_info": model_info,
-            "api_version": "1.0.0"
+            "api_version": "1.0.0",
+            "model_path": MODEL_PATH,
+            "model_exists": os.path.exists(MODEL_PATH)
         }
     except Exception as e:
         logger.error(f"Health check failed: {str(e)}")
@@ -334,6 +460,8 @@ async def test_endpoint():
         "message": "WBC Classification API is working!",
         "status": "ok",
         "model_loaded": pipeline is not None and pipeline.is_loaded if pipeline else False,
+        "model_path": MODEL_PATH,
+        "model_exists": os.path.exists(MODEL_PATH),
         "endpoints": [
             "/ - Main page",
             "/api/health - Health check",
@@ -345,38 +473,3 @@ async def test_endpoint():
     }
 
 # Main function to run the server
-def main():
-    """Main function to run the FastAPI server"""
-    
-    print("🚀 Starting WBC Classification FastAPI Server")
-    print("=" * 50)
-    print(f"📁 Model path: {MODEL_PATH}")
-    print(f"📁 Static files: {STATIC_DIR}")
-    print("📋 Available endpoints:")
-    print("   🏠 Main page: http://localhost:8000/")
-    print("   🔍 Health check: http://localhost:8000/api/health")
-    print("   📤 Single prediction: POST http://localhost:8000/api/predict")
-    print("   📤 Batch prediction: POST http://localhost:8000/api/predict_batch")
-    print("   📚 API docs: http://localhost:8000/docs")
-    print("=" * 50)
-    
-    # Check if model exists before starting
-    if not os.path.exists(MODEL_PATH):
-        print(f"⚠️  WARNING: Model file not found at {MODEL_PATH}")
-        print("   Please ensure your model file is in the Models folder")
-    
-    # Check if HTML file exists
-    html_path = os.path.join(STATIC_DIR, "index.html")
-    if not os.path.exists(html_path):
-        print(f"⚠️  WARNING: HTML file not found at {html_path}")
-        print("   Please move your index.html to the static folder")
-    
-    # Run the server
-    uvicorn.run(
-        "main:app",
-        host="0.0.0.0",
-        port=8000,
-        reload=True,
-        log_level="info"
-    )
-