Fix cache init and copy full src

Dockerfile

@@ -40,9 +40,8 @@ RUN pip uninstall -y multipart || true
 # Copy model and app code
 COPY models/vit-finetuned ./models/vit-finetuned
 
-# Copy FastAPI and Streamlit apps
-#COPY ./src/streamlit_app.py ./src/streamlit_app.py
-COPY ./src/app_gradio.py ./src/app_gradio.py
+# Copy application source files
+COPY ./src/ ./src/
 
 # Copy assets
 COPY ./images/plant-disease-logo.png ./images/plant-disease-logo.png
@@ -63,4 +62,4 @@ EXPOSE 7860
 #CMD streamlit run src/streamlit_app.py --server.port=7860 --server.address=0.0.0.0
 # To run FastAPI, comment out the above CMD and uncomment below:
 #CMD python -c "import sys, os; print('Python:', sys.version); print('Env:', dict(os.environ)); os.system('ls -lR /app');" && uvicorn src.inference.app_fastapi:app --host 0.0.0.0 --port 7860
-CMD python src/app_gradio.py
+CMD python src/app_gradio.py

src/app_gradio.py

@@ -1,13 +1,31 @@
+import argparse
+import base64
+import json
+import os
+import sys
+
+CURR_DIR = os.path.dirname(os.path.abspath(__file__))
+if CURR_DIR not in sys.path:
+    sys.path.insert(0, CURR_DIR)
+
+cache_root = os.path.abspath(os.path.join(CURR_DIR, "..", ".cache", "huggingface"))
+os.makedirs(cache_root, exist_ok=True)
+os.environ.setdefault("HF_HOME", cache_root)
+os.environ.setdefault("TRANSFORMERS_CACHE", cache_root)
+
 import gradio as gr
 from PIL import Image
 import torch
-import os
-import json
-import base64
 from transformers import AutoImageProcessor, AutoModelForImageClassification
-import sys
+
+SPACE_ENV_VARS = ("SPACE_ID", "HF_SPACE_ID", "REPO_ID")
+
 from model_paths import resolve_model_dir
 
+
+def running_in_space() -> bool:
+    return any(os.getenv(var) for var in SPACE_ENV_VARS)
+
 # Debug flag: set to True to enable, False to disable
 DEBUG = False
 
@@ -133,12 +151,15 @@ with gr.Blocks() as demo:
         gr.Markdown(f"<sub>{logo_dbg}</sub>")
 
 if __name__ == "__main__":
-    LOCAL = "--local" in sys.argv
-
-    if LOCAL:
-        # Local dev: bind to loopback so the browser check works on Windows
-        demo.launch(server_name="127.0.0.1", server_port=7860)
-    else:
-        # HF Spaces / containers: bind to 0.0.0.0 and use provided PORT
-        port = int(os.getenv("PORT", "7860"))
-        demo.launch(server_name="0.0.0.0", server_port=port)
+    parser = argparse.ArgumentParser(description="Launch the Plant Disease Classifier Gradio interface.")
+    parser.add_argument("--local", action="store_true", help="Force binding to loopback for local dev.")
+    parser.add_argument("--share", action="store_true", help="Create a public Gradio share URL.")
+    parser.add_argument("--port", type=int, help="Port to bind the Gradio server to.")
+    parser.add_argument("--host", help="Override the server host/IP address.")
+    args = parser.parse_args()
+
+    port = args.port or int(os.getenv("PORT", "7860"))
+    default_host = "127.0.0.1" if args.local or not running_in_space() else "0.0.0.0"
+    host = args.host or os.getenv("GRADIO_SERVER_NAME") or default_host
+
+    demo.launch(server_name=host, server_port=port, share=args.share)