Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,6 +1,9 @@
-import os
-import time
+import os, pathlib
+os.environ.setdefault("HOME", "/tmp")
+pathlib.Path("/tmp/.streamlit").mkdir(parents =True , exist_ok=True)
 import streamlit as st
+st.set_option("browser.gatherUsageStats", False)
+import time
 from huggingface_hub import InferenceClient
 import re
 import gspread
@@ -45,7 +48,9 @@ class ConfigManager:
         self.google_creds_json = os.getenv("GOOGLE_SHEETS_CREDENTIALS")
         self.google_sheets_id = os.getenv("GOOGLE_SHEETS_ID")
 
-    
+        
+
+
         missing_vars = []
 
     def get_google_client(self):
@@ -127,13 +132,16 @@ class GoogleSheetsLogger:
 
 
 class AIAssistant:
-    def __init__(self, config: ConfigManager):
-        self.model = os.environ.get("HF_MODEL", "mistralai/Mistral-7B-Instruct-v0.2")
+    def __init__(self, config):
+        self.model =  os.getenv("HF_MODEL",  "HuggingFaceH4/zephyr-7b-beta") 
+        token = os.getenv("HF_TOKEN")
+        if os.getenv("SPACE_ID") or os.getenv("HF_SPACE_ID"):
+            token =  None 
+
         self.client = InferenceClient(
             model=self.model,
-            token=config.hf_token,   
+            token=token,
             timeout=60.0,
-            provider="hf-inference",    
         )
 
         self.base_prompt = f"""
@@ -846,4 +854,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()