Refactor Dockerfile and main.py to set working directory and PYTHONPATH correctly; ensure imports function as expected

Dockerfile

@@ -2,10 +2,10 @@
 FROM python:3.10-slim
 
 # Set working directory
-WORKDIR /
+WORKDIR /app
 
 # Set PYTHONPATH to ensure imports work correctly
-ENV PYTHONPATH=/
+ENV PYTHONPATH=/app
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
@@ -31,5 +31,6 @@ ENV RELOAD=False
 
 # Run the application
 # Hugging Face Spaces sets PORT environment variable automatically
-CMD uvicorn main:app --host 0.0.0.0 --port ${PORT:-7860}
+# Note: WORKDIR is already set to /app above, and main.py adds /app to sys.path
+CMD ["sh", "-c", "uvicorn main:app --host 0.0.0.0 --port ${PORT:-7860}"]
 

main.py

@@ -1,5 +1,13 @@
 """Main FastAPI application entry point"""
 
+import sys
+from pathlib import Path
+
+# Add the app directory to Python path to ensure imports work
+app_dir = Path(__file__).parent
+if str(app_dir) not in sys.path:
+    sys.path.insert(0, str(app_dir))
+
 from contextlib import asynccontextmanager
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware