Upload utils/gemma_translation.py with huggingface_hub

utils/gemma_translation.py

@@ -16,7 +16,8 @@ import contextlib
 
 # Import configuration defaults
 from config import DEFAULT_CONFIG
-
+from utils.model_bootstrap import ensure_gemma
+MODEL_PATH = ensure_gemma()
 
 @contextlib.contextmanager
 def suppress_stdout_stderr():
@@ -49,7 +50,8 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # Model configuration from config
-ORIGINAL_MODEL_PATH = os.path.join("local_llms", "gemma-3-12b-it-Q4_K_M.gguf")
+# ORIGINAL_MODEL_PATH = os.path.join("local_llms", "gemma-3-12b-it-Q4_K_M.gguf")
+ORIGINAL_MODEL_PATH = MODEL_PATH
 MODEL_DIR = os.path.join("local_llms", "instances")
 os.makedirs(MODEL_DIR, exist_ok=True)
 
@@ -123,28 +125,12 @@ class GemmaTranslator:
         
         # Create a session-specific model path
         self.model_path = self._get_session_model_path()
-    
+
     def _get_session_model_path(self):
-        """Get or create a session-specific model file."""
-        
-        session_model_filename = f"gemma-{self.session_id}.gguf"
-        session_model_path = os.path.join(MODEL_DIR, session_model_filename)
-        
-        # If the model file doesn't exist yet, create it by copying the original
-        if not os.path.exists(session_model_path):
-            if not os.path.exists(ORIGINAL_MODEL_PATH):
-                raise FileNotFoundError(f"Original model file not found: {ORIGINAL_MODEL_PATH}")
-                
-            logger.info(f"Creating session-specific model file for {self.session_id}")
-            try:
-                shutil.copy2(ORIGINAL_MODEL_PATH, session_model_path)
-                logger.info(f"Created session model at {session_model_path}")
-            except Exception as e:
-                logger.error(f"Failed to create session model: {str(e)}")
-                # Fallback to original model if copy fails
-                return ORIGINAL_MODEL_PATH
-                
-        return session_model_path
+        """Use the single shared GGUF file instead of copying per session."""
+        if not os.path.exists(ORIGINAL_MODEL_PATH):
+            raise FileNotFoundError(f"Original model file not found: {ORIGINAL_MODEL_PATH}")
+        return ORIGINAL_MODEL_PATH
     
     
     def load_model(self,