Upload folder using huggingface_hub

web/routes/websocket.py

@@ -98,8 +98,13 @@ async def websocket_chat(websocket: WebSocket):
                 # Get session for this connection (auto-recreate if lost)
                 session = get_session(connection_id)
                 if not session:
-                    logger.warning(f"Session lost for {connection_id[:8]}, recreating...")
-                    session = create_session(connection_id)
+                    logger.warning(f"Session lost for {connection_id[:8]}, requesting keys...")
+                    # Ask client to resend keys (e.g., after container restart)
+                    await manager.send_json(websocket, {
+                        "type": "request_keys",
+                        "reason": "Session expired, please reconnect."
+                    })
+                    continue
 
                 # Callback for streaming
                 async def stream_callback(event_type: str, content: str, **kwargs):

web/static/js/chat.js

@@ -385,6 +385,7 @@ class EurusChat {
         if (!message || !this.isConnected) return;
 
         this.addUserMessage(message);
+        this._lastSentMessage = message;
         this.ws.send(JSON.stringify({ message }));
 
         this.messageInput.value = '';
@@ -406,6 +407,22 @@ class EurusChat {
                 }
                 break;
 
+            case 'request_keys':
+                // Server lost our session (e.g., container restart) — resend keys
+                console.warn('Server requested keys:', data.reason);
+                this.removeThinkingIndicator();
+                this.autoSendSessionKeys();
+                // Retry the last message after a short delay for session init
+                if (this._lastSentMessage) {
+                    setTimeout(() => {
+                        if (this.ws && this.ws.readyState === WebSocket.OPEN) {
+                            this.ws.send(JSON.stringify({ message: this._lastSentMessage }));
+                            this.showThinkingIndicator();
+                        }
+                    }, 3000);
+                }
+                break;
+
             case 'thinking':
                 this.showThinkingIndicator();
                 break;

web/templates/index.html

@@ -32,6 +32,12 @@
                 <li>"Show me SST for California coast, Jan 2024"</li>
                 <li>"Plot temperature in the Gulf of Mexico"</li>
             </ul>
+            <p style="margin-top: 10px; font-size: 0.85em; opacity: 0.7;">
+                ⚠️ <strong>Experimental</strong> — This is a research prototype. Please avoid requesting very large
+                datasets
+                (e.g., global multi-year downloads), especially via this HF Space. For heavy workloads, use the
+                📦 Arraylake Code buttons to access data directly.
+            </p>
         </div>
     </div>