disable websocket2

Dockerfile

@@ -56,6 +56,9 @@ RUN mkdir -p /home/user/app/.files && \
     mkdir -p /home/user/.local && \
     chown -R user:user /home/user
 
+RUN chown -R user:user /home/user && \
+    chmod -R 755 /home/user/app
+
 RUN pip install pydantic==2.10.1 chainlit
 
 COPY chainlit.md .

app.py

@@ -39,7 +39,11 @@ from tools import search_pypi, write_to_docx
 from prompts import describe_imports, main_prompt, documenter_prompt
 from states import AgentState
 
-os.environ["CHAINLIT_DISABLE_WEBSOCKETS"] = "true"
+if os.environ.get("SPACE_ID"):  # Check if running on HF Spaces
+    os.environ["CHAINLIT_DISABLE_WEBSOCKETS"] = "true"
+    # Also consider setting these for HF Spaces
+    os.environ["CHAINLIT_SERVER_PORT"] = "7860"
+    os.environ["CHAINLIT_SERVER_HOST"] = "0.0.0.0"
 
 
 # Global variables to store processed data
@@ -51,6 +55,8 @@ qdrant_client = None
 
 @cl.on_chat_start
 async def on_chat_start():
+    print("Chat session started")
+
     await cl.Message(content="Welcome to the Python Code Documentation Assistant! Please upload a Python file to get started.").send()
 
 @cl.on_message