idek

Dockerfile

@@ -1,10 +1,30 @@
-FROM python:3.10
+FROM python:3.10-slim
+
+# Set environment variables
+ENV PYTHONUNBUFFERED=1
+ENV HF_HOME=/tmp
 
 WORKDIR /code
 
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy and install Python dependencies
 COPY requirements.txt .
+RUN pip install --no-cache-dir --upgrade pip
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Copy application code
 COPY . .
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Expose port
+EXPOSE 7860
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=10s --start-period=30s --retries=3 \
+    CMD curl -f http://localhost:7860/ || exit 1
+
+# Run the application
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--log-level", "info"]

requirements.txt

@@ -1,6 +1,7 @@
 fastapi
-uvicorn
+uvicorn[standard]
 transformers
-torch
-pydantic>=1.10,<2
+torch --index-url https://download.pytorch.org/whl/cpu
+pydantic>=2.0
 huggingface_hub
+requests

test_local.py

@@ -0,0 +1,62 @@
+"""
+Test script to verify FastAPI app works locally
+"""
+import subprocess
+import time
+import requests
+import threading
+import sys
+
+def start_server():
+    """Start the FastAPI server in background"""
+    try:
+        subprocess.run([
+            sys.executable, "-m", "uvicorn", 
+            "app:app", "--host", "127.0.0.1", "--port", "8000"
+        ], check=False)
+    except Exception as e:
+        print(f"Server failed to start: {e}")
+
+def test_endpoints():
+    """Test the API endpoints"""
+    base_url = "http://127.0.0.1:8000"
+    
+    # Wait a moment for server to start
+    time.sleep(3)
+    
+    try:
+        # Test health endpoint
+        print("🔹 Testing health endpoint...")
+        health_resp = requests.get(base_url, timeout=5)
+        print(f"Health Status: {health_resp.status_code}")
+        if health_resp.status_code == 200:
+            print(f"Health Response: {health_resp.json()}")
+        else:
+            print(f"Health Error: {health_resp.text}")
+        
+        # Test prediction endpoint
+        print("\n🔹 Testing prediction endpoint...")
+        pred_resp = requests.post(
+            base_url,
+            json={"text": "Congratulations! You've won a free cruise!"},
+            headers={"Content-Type": "application/json"},
+            timeout=10
+        )
+        print(f"Prediction Status: {pred_resp.status_code}")
+        if pred_resp.status_code == 200:
+            print(f"Prediction Response: {pred_resp.json()}")
+        else:
+            print(f"Prediction Error: {pred_resp.text}")
+            
+    except Exception as e:
+        print(f"Test failed: {e}")
+
+if __name__ == "__main__":
+    print("Starting FastAPI server...")
+    
+    # Start server in a separate thread
+    server_thread = threading.Thread(target=start_server, daemon=True)
+    server_thread.start()
+    
+    # Run tests
+    test_endpoints()