add docker files

Dockerfile

@@ -0,0 +1,9 @@
+FROM python:3.10.9
+
+COPY . .
+
+WORKDIR /
+
+RUN pip install --no-cache-dir --upgrade -r /requirements.txt
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,9 +1,11 @@
 from fastapi import FastAPI, Request
 from fastapi.templating import Jinja2Templates
 from fastapi.staticfiles import StaticFiles
-from fastapi.responses import HTMLResponse
+from fastapi.responses import HTMLResponse, JSONResponse
 import pandas as pd
 from pathlib import Path
+import asyncio
+import time
 
 from app.backend.constant import Navigation, ModelProvider, EvaluationMetric, EmbdDtype, EmbdDim, Similarity
 from app.backend.data_engine import DataEngine
@@ -17,45 +19,93 @@ app.mount("/static", StaticFiles(directory="app/static"), name="static")
 templates = Jinja2Templates(directory="app/templates")
 
 # Initialize data engine
-data_engine = DataEngine()
+data_engine = None
+initialization_start_time = None
+
+async def initialize_data_engine():
+    global data_engine, initialization_start_time
+    if data_engine is None and initialization_start_time is None:
+        initialization_start_time = time.time()
+        try:
+            data_engine = DataEngine()
+            print("Successfully initialized DataEngine")
+        except Exception as e:
+            print(f"Error initializing DataEngine: {str(e)}")
+        finally:
+            initialization_start_time = None
+
+@app.on_event("startup")
+async def startup_event():
+    asyncio.create_task(initialize_data_engine())
+
+@app.get("/health")
+async def health():
+    if data_engine is not None:
+        return {"status": "healthy", "initialized": True}
+    elif initialization_start_time is not None:
+        return {"status": "initializing", "initialized": False, "elapsed_time": time.time() - initialization_start_time}
+    else:
+        return {"status": "not_initialized", "initialized": False}
 
 @app.get("/", response_class=HTMLResponse)
 async def home(request: Request, tab: str = "text"):
-    # Get initial data based on tab
-    if tab == "multimodal":
-        data = data_engine.get_filtered_data(navigation="multimodal")
-    else:
-        data = data_engine.get_data()
+    if data_engine is None:
+        return templates.TemplateResponse("loading.html", {"request": request})
     
-    # Convert data to list for template rendering
-    data_list = data.values.tolist()
-    columns = data.columns.tolist()
-
-    return templates.TemplateResponse("index.html", {
-        "request": request,
-        "active_tab": tab,
-        "data": data_list,
-        "columns": columns,
-        "navigations": [e.value for e in Navigation],
-        "embd_types": [e.value for e in EmbdDtype],
-        "embd_dims": [e.value for e in EmbdDim],
-        "similarities": [e.value for e in Similarity]
-    })
+    try:
+        # Get initial data based on tab
+        if tab == "multimodal":
+            data = data_engine.get_filtered_data(navigation="multimodal")
+        else:
+            data = data_engine.get_data()
+        
+        # Convert data to list for template rendering
+        data_list = data.values.tolist()
+        columns = data.columns.tolist()
+
+        return templates.TemplateResponse("index.html", {
+            "request": request,
+            "active_tab": tab,
+            "data": data_list,
+            "columns": columns,
+            "navigations": [e.value for e in Navigation],
+            "embd_types": [e.value for e in EmbdDtype],
+            "embd_dims": [e.value for e in EmbdDim],
+            "similarities": [e.value for e in Similarity]
+        })
+    except Exception as e:
+        print(f"Error in home route: {str(e)}")
+        return templates.TemplateResponse("error.html", {
+            "request": request,
+            "error": str(e)
+        })
 
 @app.post("/api/filter")
 async def filter_data(filters: dict):
-    # Apply filters to data
-    data = data_engine.get_filtered_data(
-        navigation=filters.get('navigation'),
-        embd_type=filters.get('embd_type'),
-        embd_dims=filters.get('embd_dims'),
-        similarity=filters.get('similarity')
-    )
+    if data_engine is None:
+        return JSONResponse(
+            status_code=503,
+            content={"error": "Service is still initializing"}
+        )
     
-    # Convert to list for JSON response
-    return data.values.tolist()
-
+    try:
+        # Apply filters to data
+        data = data_engine.get_filtered_data(
+            navigation=filters.get('navigation'),
+            embd_type=filters.get('embd_type'),
+            embd_dims=filters.get('embd_dims'),
+            similarity=filters.get('similarity')
+        )
+        
+        # Convert to list for JSON response
+        return data.values.tolist()
+    except Exception as e:
+        print(f"Error in filter route: {str(e)}")
+        return JSONResponse(
+            status_code=500,
+            content={"error": str(e)}
+        )
 
-if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7890)
+# if __name__ == "__main__":
+#     import uvicorn
+#     uvicorn.run(app, host="0.0.0.0", port=7891)

app/templates/error.html

@@ -0,0 +1,59 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>MTEB Leaderboard - Error</title>
+    <!-- Semi Design CSS -->
+    <link rel="stylesheet" href="https://unpkg.com/@douyinfe/semi-ui/dist/css/semi.min.css">
+    <!-- Custom CSS -->
+    <link rel="stylesheet" href="{{ url_for('static', path='css/style.css') }}">
+    <style>
+        .error-container {
+            display: flex;
+            flex-direction: column;
+            align-items: center;
+            justify-content: center;
+            min-height: 100vh;
+            text-align: center;
+            padding: 20px;
+        }
+        
+        .error-icon {
+            font-size: 48px;
+            color: #e74c3c;
+            margin: 20px 0;
+        }
+        
+        .error-message {
+            font-size: 1.2em;
+            margin: 10px 0;
+            color: #666;
+        }
+        
+        .retry-button {
+            margin-top: 20px;
+            padding: 10px 20px;
+            background-color: #3498db;
+            color: white;
+            border: none;
+            border-radius: 4px;
+            cursor: pointer;
+            text-decoration: none;
+        }
+        
+        .retry-button:hover {
+            background-color: #2980b9;
+        }
+    </style>
+</head>
+<body>
+    <div class="error-container">
+        <h1>MTEB Leaderboard</h1>
+        <div class="error-icon">⚠️</div>
+        <h2>An Error Occurred</h2>
+        <p class="error-message">{{ error }}</p>
+        <a href="/" class="retry-button">Retry</a>
+    </div>
+</body>
+</html>

app/templates/loading.html

@@ -0,0 +1,81 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>MTEB Leaderboard - Loading</title>
+    <!-- Semi Design CSS -->
+    <link rel="stylesheet" href="https://unpkg.com/@douyinfe/semi-ui/dist/css/semi.min.css">
+    <!-- Custom CSS -->
+    <link rel="stylesheet" href="{{ url_for('static', path='css/style.css') }}">
+    <style>
+        .loading-container {
+            display: flex;
+            flex-direction: column;
+            align-items: center;
+            justify-content: center;
+            min-height: 100vh;
+            text-align: center;
+            padding: 20px;
+        }
+        
+        .loading-spinner {
+            border: 4px solid #f3f3f3;
+            border-top: 4px solid #3498db;
+            border-radius: 50%;
+            width: 50px;
+            height: 50px;
+            animation: spin 1s linear infinite;
+            margin: 20px 0;
+        }
+        
+        @keyframes spin {
+            0% { transform: rotate(0deg); }
+            100% { transform: rotate(360deg); }
+        }
+        
+        .loading-text {
+            font-size: 1.2em;
+            margin: 10px 0;
+        }
+        
+        #loading-status {
+            color: #666;
+            margin-top: 10px;
+        }
+    </style>
+</head>
+<body>
+    <div class="loading-container">
+        <h1>MTEB Leaderboard</h1>
+        <div class="loading-spinner"></div>
+        <p class="loading-text">Loading data...</p>
+        <p id="loading-status">Please wait while we initialize the service</p>
+    </div>
+
+    <script>
+        async function checkStatus() {
+            try {
+                const response = await fetch('/health');
+                const data = await response.json();
+                
+                if (data.initialized) {
+                    window.location.reload();
+                } else {
+                    const status = document.getElementById('loading-status');
+                    if (data.status === 'initializing') {
+                        status.textContent = `Initializing... (${Math.round(data.elapsed_time)}s)`;
+                    } else {
+                        status.textContent = 'Waiting for initialization...';
+                    }
+                }
+            } catch (error) {
+                console.error('Error checking status:', error);
+            }
+        }
+
+        // Check status every 2 seconds
+        setInterval(checkStatus, 2000);
+    </script>
+</body>
+</html>