Updated files

main.py

@@ -88,12 +88,14 @@ async def initialize_research_mate():
     print("🚀 Starting ResearchMate background initialization...")
     
     try:
-        # Ensure vectorstore/chroma persist directory exists before initializing components
-        base_dir = Path(__file__).parent.resolve()
-        chroma_dir = base_dir / "tmp" / "researchmate" / "chroma_persist"
+        # Use /data on Hugging Face Spaces, else use local project-relative path
+        running_on_hf = os.environ.get("HF_SPACE") == "1" or os.environ.get("SPACE_ID")
+        if running_on_hf:
+            chroma_dir = Path("/data/researchmate/chroma_persist")
+        else:
+            base_dir = Path(__file__).parent.resolve()
+            chroma_dir = base_dir / "tmp" / "researchmate" / "chroma_persist"
         chroma_dir.mkdir(parents=True, exist_ok=True)
-
-        # Set environment variable for ChromaDB persist directory (if needed by your code)
         os.environ["CHROMA_PERSIST_DIR"] = str(chroma_dir)
 
         # Run initialization in thread pool to avoid blocking

src/components/config.py

@@ -88,8 +88,13 @@ class Config:
                 print(f"Created/verified directory: {directory}")
             except PermissionError as e:
                 print(f"Permission error creating {directory}: {e}")
-                # Try to create in /tmp as fallback
-                fallback_dir = f"/tmp/researchmate/{Path(directory).name}"
+                # Try to create in /data as fallback on Hugging Face, else ./tmp
+                import os
+                running_on_hf = os.environ.get("HF_SPACE") == "1" or os.environ.get("SPACE_ID")
+                if running_on_hf:
+                    fallback_dir = f"/data/researchmate/{Path(directory).name}"
+                else:
+                    fallback_dir = f"./tmp/researchmate/{Path(directory).name}"
                 try:
                     Path(fallback_dir).mkdir(parents=True, exist_ok=True)
                     print(f"Created fallback directory: {fallback_dir}")

src/settings.py

@@ -51,7 +51,14 @@ class UploadConfig:
     max_file_size: int = 50 * 1024 * 1024  # 50MB
     allowed_extensions: List[str] = None
     upload_directory: str = "./uploads"
-    temp_directory: str = "./tmp"
+    import os
+    @property
+    def temp_directory(self):
+        running_on_hf = os.environ.get("HF_SPACE") == "1" or os.environ.get("SPACE_ID")
+        if running_on_hf:
+            return "/data/tmp"
+        else:
+            return "./tmp"
     
     def __post_init__(self):
         if self.allowed_extensions is None:
@@ -118,7 +125,11 @@ class Settings:
     def _get_default_config_file(self) -> str:
         """Get default configuration file path"""
         # Use writable config directory
-        config_dir = os.environ.get('CONFIG_DIR', '/tmp/researchmate/config')
+        running_on_hf = os.environ.get("HF_SPACE") == "1" or os.environ.get("SPACE_ID")
+        if running_on_hf:
+            config_dir = os.environ.get('CONFIG_DIR', '/data/researchmate/config')
+        else:
+            config_dir = os.environ.get('CONFIG_DIR', './tmp/researchmate/config')
         return str(Path(config_dir) / "settings.json")
     
     def _load_config(self):
@@ -156,7 +167,11 @@ class Settings:
         self.server.log_level = os.getenv("LOG_LEVEL", self.server.log_level)
         
         # Database configuration - USE ENVIRONMENT VARIABLE
-        self.database.chroma_persist_dir = os.getenv("CHROMA_DIR", "/tmp/researchmate/chroma_persist")
+        running_on_hf = os.environ.get("HF_SPACE") == "1" or os.environ.get("SPACE_ID")
+        if running_on_hf:
+            self.database.chroma_persist_dir = os.getenv("CHROMA_DIR", "/data/researchmate/chroma_persist")
+        else:
+            self.database.chroma_persist_dir = os.getenv("CHROMA_DIR", "./tmp/researchmate/chroma_persist")
         self.database.collection_name = os.getenv("COLLECTION_NAME", self.database.collection_name)
         self.database.similarity_threshold = float(os.getenv("SIMILARITY_THRESHOLD", self.database.similarity_threshold))
         self.database.max_results = int(os.getenv("MAX_RESULTS", self.database.max_results))
@@ -169,12 +184,24 @@ class Settings:
         
         # Upload configuration - USE ENVIRONMENT VARIABLES
         self.upload.max_file_size = int(os.getenv("MAX_FILE_SIZE", self.upload.max_file_size))
-        self.upload.upload_directory = os.getenv("UPLOADS_DIR", "/tmp/researchmate/uploads")
-        self.upload.temp_directory = os.getenv("TEMP_DIR", "/tmp/researchmate/tmp")
+        running_on_hf = os.environ.get("HF_SPACE") == "1" or os.environ.get("SPACE_ID")
+        if running_on_hf:
+            self.upload.upload_directory = os.getenv("UPLOADS_DIR", "/data/researchmate/uploads")
+        else:
+            self.upload.upload_directory = os.getenv("UPLOADS_DIR", "./tmp/researchmate/uploads")
+        running_on_hf = os.environ.get("HF_SPACE") == "1" or os.environ.get("SPACE_ID")
+        if running_on_hf:
+            self.upload.temp_directory = os.getenv("TEMP_DIR", "/data/researchmate/tmp")
+        else:
+            self.upload.temp_directory = os.getenv("TEMP_DIR", "./tmp/researchmate/tmp")
         
         # Logging configuration - USE ENVIRONMENT VARIABLE
         self.logging.level = os.getenv("LOG_LEVEL", self.logging.level)
-        self.logging.file_path = os.getenv("LOG_FILE", os.path.join(os.getenv("LOGS_DIR", "/tmp/researchmate/logs"), "app.log"))
+        running_on_hf = os.environ.get("HF_SPACE") == "1" or os.environ.get("SPACE_ID")
+        if running_on_hf:
+            self.logging.file_path = os.getenv("LOG_FILE", os.path.join(os.getenv("LOGS_DIR", "/data/researchmate/logs"), "app.log"))
+        else:
+            self.logging.file_path = os.getenv("LOG_FILE", os.path.join(os.getenv("LOGS_DIR", "./tmp/researchmate/logs"), "app.log"))
     
     def _validate_config(self):
         """Validate configuration settings"""