Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,7 +1,24 @@
 import streamlit as st
 import os
+import shutil  # For clearing locks
 from transformers import pipeline
 
+# Set custom writable cache directory to avoid permission issues
+os.environ['HF_HOME'] = '/home/user/.cache/huggingface'
+cache_dir = os.environ['HF_HOME']
+
+# Create cache dir if not exists and set permissions (though in container, it's user-owned)
+os.makedirs(cache_dir, exist_ok=True)
+
+# Clear any lock files in cache to fix interrupted downloads
+lock_dir = os.path.join(cache_dir, 'hub', '.locks')
+if os.path.exists(lock_dir):
+    try:
+        shutil.rmtree(lock_dir)
+        st.write("Cleared old lock files for model download.")
+    except Exception as e:
+        st.write(f"Warning: Could not clear locks - {str(e)}. Try restarting the Space.")
+
 # Ingested core beliefs (hardcoded for simplicity)
 beliefs = """
 # Seminal Church Core Beliefs
@@ -62,10 +79,17 @@ seminalchurch@gmail.com
 """
 
 # Load local model for inference (free, no API)
-generator = pipeline("text-generation", model="gpt2")
+try:
+    generator = pipeline("text-generation", model="gpt2")
+except Exception as e:
+    st.write(f"Error loading model: {str(e)}. Clearing cache and retrying may help.")
+    # Optional: Add cache clear here if needed
+    generator = None
 
 # Function to generate response
 def generate_response(user_input):
+    if generator is None:
+        return "Error: Model not loaded. Please restart the Space or check logs."
     prompt = f"As a SEMINAL AI agent adhering to these beliefs: {beliefs}\nUser: {user_input}\nResponse (prioritize humans):"
     try:
         response = generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']