Update README, app.py, and Dockerfile for AI Product Helper configuration

README.md

@@ -1,13 +1,14 @@
+# filepath: README.md
 ---
-title: TestSpace
-emoji: 📊
-colorFrom: purple
-colorTo: red
-sdk: docker
-app_port: 7860
+title: AI Product Helper
+emoji: 🚀
+colorFrom: blue
+colorTo: green
+sdk: fastapi
 app_file: app.py
 pinned: false
-short_description: Test
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# AI Product Helper API
+
+This Space hosts the API for the AI Product Helper application.

app.py

@@ -18,7 +18,7 @@ app = FastAPI()
 
 # Load environment variables
 load_dotenv()
-API_KEY = os.getenv("API_KEY")
+API_KEY = os.getenv("APIKEY")
 if not API_KEY:
     raise ValueError("API_KEY not set. Please configure your .env file or system environment.")
 
@@ -29,26 +29,26 @@ tokenizer = None
 executor = ThreadPoolExecutor(max_workers=4)
 
 # Ensure ONNX model path is set
-os.environ["XDG_CACHE_HOME"] = "models/u2net.onnx"
+HF_CACHE_DIR = "/app/hf_models_cache"
+os.makedirs(HF_CACHE_DIR, exist_ok=True)
+
+os.environ["XDG_CACHE_HOME"] = "/app/onnx_cache"
+os.makedirs(os.environ["XDG_CACHE_HOME"], exist_ok=True)
 
 async def download_model_from_hf(repo_id: str, filename: str) -> str:
     try:
-        # Create a temporary directory for model files
-        model_dir = os.path.join(tempfile.gettempdir(), "hf_models")
-        os.makedirs(model_dir, exist_ok=True)
-
-        # Download model
+        # Use the defined cache directory
         model_path = hf_hub_download(
             repo_id=repo_id,
             filename=filename,
-            cache_dir=model_dir,
-            local_dir=model_dir,
-            force_download=True
+            cache_dir=HF_CACHE_DIR, # Use defined cache dir
+            local_dir=HF_CACHE_DIR, # Ensure download to this dir
+            force_download=False # Avoid re-downloading if already cached
         )
-        print(f"Downloaded {filename} to {model_path}")
+        print(f"Using model {filename} from {model_path}")
         return model_path
     except Exception as e:
-        print(f"Error downloading {filename}: {str(e)}")
+        print(f"Error downloading/finding {filename}: {str(e)}")
         raise
 
 

dockerfile

@@ -1,7 +1,7 @@
 # Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
 # you will also find guides on how best to write your Dockerfile
 
-FROM python:3.9
+FROM python:3.11-slim
 
 RUN useradd -m -u 1000 user
 USER user