Update the dockerfile and few updates in main.py for error and health check

.dockerignore

@@ -0,0 +1,44 @@
+# Python
+__pycache__/
+*.py[cod]
+*.pyo
+*.pyd
+*.pdb
+*.egg-info/
+*.eggs/
+*.log
+
+# Virtual environments
+venv/
+env/
+.venv/
+.env/
+
+# OS files
+.DS_Store
+Thumbs.db
+
+# Git
+.git/
+.gitignore
+
+# Build artifacts
+*.whl
+dist/
+build/
+
+# IDE/editor config
+.vscode/
+.idea/
+*.sublime-project
+*.sublime-workspace
+
+# Docker
+Dockerfile
+docker-compose.yml
+*.tar
+
+# Misc
+*.zip
+*.bak
+*.tmp

Dockerfile

@@ -20,3 +20,8 @@ COPY --chown=user . /
 
 # Run FastAPI app with Uvicorn on port 7860
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+
+# Add this before `CMD` in Dockerfile
+RUN python -c "from transformers import pipeline; pipeline('zero-shot-classification', model='joeddav/xlm-roberta-large-xnli')"
+RUN python -c "from transformers import AutoTokenizer, AutoModelForSeq2SeqLM; AutoTokenizer.from_pretrained('google/flan-t5-base'); AutoModelForSeq2SeqLM.from_pretrained('google/flan-t5-base')"
+

main.py

@@ -10,7 +10,19 @@ from dateparser.search import search_dates
 import uuid
 import time
 
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
+from fastapi.requests import Request
+from fastapi import status
+
 app = FastAPI()
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # or your domain(s)
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 
 # Load classification and summarization models
 classifier = pipeline("zero-shot-classification", model="joeddav/xlm-roberta-large-xnli")
@@ -229,7 +241,15 @@ def get_meta_info(text: str):
 
 @app.get("/health")
 def health_check():
-    return {"status": "ok", "message": "API is healthy 🚀"}
+    return {"message": "✅ Hello from yourpartner/demospace — API is running!"}
+
+@app.exception_handler(404)
+async def not_found_handler(request: Request, exc):
+    return JSONResponse(status_code=404, content={"error": "Route not found"})
+
+@app.exception_handler(500)
+async def internal_error_handler(request: Request, exc):
+    return JSONResponse(status_code=500, content={"error": "Internal server error"})
 
 @app.post("/analyze")
 async def analyze(input: TextInput):

requirements.txt

@@ -6,4 +6,6 @@ dateparser==1.2.0
 # spacy
 langdetect
 textblob
-sentencepiece
+sentencepiece
+protobuf
+scikit-learn