Fixed Display issues

Caddyfile

@@ -1,16 +1,22 @@
 :7860 {
     # Route WebSocket streams (iii-stream on 3112)
     handle /stream/* {
-        reverse_proxy localhost:3112
+        reverse_proxy localhost:3112 {
+            header_up Host localhost:3112
+        }
     }
 
     # Route REST API endpoints (iii-http on 3111)
     handle /agentmemory/* {
-        reverse_proxy localhost:3111
+        reverse_proxy localhost:3111 {
+            header_up Host localhost:3111
+        }
     }
 
     # Route Viewer dashboard (on 3113)
     handle {
-        reverse_proxy localhost:3113
+        reverse_proxy localhost:3113 {
+            header_up Host localhost:3113
+        }
     }
 }

start.sh

@@ -3,17 +3,34 @@
 # Create config folder
 mkdir -p ~/.agentmemory
 
+# The daemon runs inside the Space and must talk to its own services on local
+# container ports. Use the public hf.space URL only from your local machine.
+export AGENTMEMORY_URL=http://localhost:3111
+export III_ENGINE_URL=ws://localhost:49134
+
+# Hugging Face proxies requests with the public *.hf.space host. agentmemory's
+# viewer rejects unknown Host headers unless they are explicitly allowed.
+if [ -n "${SPACE_HOST}" ]; then
+  export VIEWER_ALLOWED_HOSTS="${VIEWER_ALLOWED_HOSTS:-${SPACE_HOST},${SPACE_HOST}:443,${SPACE_HOST}:7860}"
+  export VIEWER_ALLOWED_ORIGINS="${VIEWER_ALLOWED_ORIGINS:-https://${SPACE_HOST},http://${SPACE_HOST},http://localhost:3111,http://localhost:3113,http://127.0.0.1:3111,http://127.0.0.1:3113}"
+fi
+
 # Dynamically write env config from environment variables
 cat <<EOF > ~/.agentmemory/.env
 # =============================================================================
 # agentmemory configuration (Hugging Face Spaces)
 # =============================================================================
 GEMINI_API_KEY=${GEMINI_API_KEY}
+AGENTMEMORY_SECRET=${AGENTMEMORY_SECRET}
+AGENTMEMORY_URL=${AGENTMEMORY_URL}
+III_ENGINE_URL=${III_ENGINE_URL}
 GEMINI_MODEL=gemini-2.5-flash
 EMBEDDING_PROVIDER=gemini
 CONSOLIDATION_ENABLED=true
 GRAPH_EXTRACTION_ENABLED=true
 AGENTMEMORY_REFLECT=true
+VIEWER_ALLOWED_HOSTS=${VIEWER_ALLOWED_HOSTS}
+VIEWER_ALLOWED_ORIGINS=${VIEWER_ALLOWED_ORIGINS}
 EOF
 
 # Start agentmemory daemon in the background