Update app.py

app.py

@@ -33,6 +33,22 @@ LOGGING_CONFIG = {
         'process_query': True
     }
 }
+def download_file_with_progress(url: str, filename: str):
+    """Download a file with progress bar using requests"""
+    response = requests.get(url, stream=True)
+    total_size = int(response.headers.get('content-length, 0))
+    
+    with open(filename, 'wb') as file, tqdm(
+        desc=filename,
+        total=total_size,
+        unit='iB',
+        unit_scale=True,
+        unit_divisor=1024,
+    ) as progress_bar:
+        for data in response.iter_content(chunk_size=1024):
+            size = file.write(data)
+            progress_bar.update(size)
+            
 @st.cache_data
 def load_from_drive(file_id: str):
     """Load pickle file directly from Google Drive"""
@@ -120,13 +136,20 @@ def check_environment():
 
 @st.cache_resource
 def initialize_model():
-    """Initialize the Llama model once"""
-    #model_path = "mistral-7b-v0.1.Q4_K_M.gguf"
-    model_path = "neuralmagic/Mistral-7B-Instruct-v0.3-GPTQ-4bit"
-    # model_path = "TheBloke/Mistral-7B-v0.1-GGUF"
-    if not os.path.exists(model_path):
-        st.error(f"Model file {model_path} not found!")
-        st.stop()
+    """Initialize the model with proper error handling and verification"""
+    try:
+        if not os.path.exists(self.model_path):
+            direct_url = "https://huggingface.co/TheBloke/Mistral-7B-v0.1-GGUF/resolve/main/mistral-7b-v0.1.Q4_K_M.gguf"
+            download_file_with_progress(direct_url, self.model_path)
+
+        # Verify file exists and has content
+        if not os.path.exists(self.model_path):
+            raise FileNotFoundError(f"Model file {self.model_path} not found after download attempts")
+            
+        if os.path.getsize(self.model_path) < 1000000:  # Less than 1MB
+            os.remove(self.model_path)
+            raise ValueError("Downloaded model file is too small, likely corrupted")
+
     
     llm_config = {
         "n_ctx": 2048,