Update app.py

app.py

@@ -204,15 +204,37 @@ document_inspection_tool = None
 def get_user_session(request: gr.Request) -> str:
     """Get or create a unique session ID for the user."""
     if not request:
+        logger.warning("No request object, using random session ID")
         return str(uuid.uuid4())
     
     # Try to get session from headers or create new one
     session_id = request.headers.get("x-session-id")
     if not session_id:
-        session_id = str(uuid.uuid4())
+        # Use client IP and user agent as a more stable identifier
+        client_ip = request.client.host if hasattr(request, 'client') and request.client else "unknown"
+        user_agent = request.headers.get("user-agent", "unknown")
+        # Create a hash-based session ID for more stability
+        import hashlib
+        session_hash = hashlib.md5(f"{client_ip}:{user_agent}".encode()).hexdigest()
+        session_id = f"session_{session_hash[:8]}"
+        logger.info(f"Created stable session ID {session_id} for client {client_ip}")
     
     return session_id
 
+def get_stable_session_id(request: gr.Request) -> str:
+    """Get a stable session ID that persists across requests."""
+    if not request:
+        return "default_session"
+    
+    # Use a combination of client info for more stable sessions
+    client_ip = getattr(request.client, 'host', 'unknown') if request.client else 'unknown'
+    user_agent = request.headers.get("user-agent", "unknown")
+    
+    # Create a stable session ID
+    import hashlib
+    session_hash = hashlib.md5(f"{client_ip}:{user_agent}".encode()).hexdigest()
+    return f"user_{session_hash[:12]}"
+
 def get_session_data(session_id: str) -> dict:
     """Get session data for a specific user."""
     with session_lock:
@@ -296,10 +318,12 @@ class GradioUI:
     def interact_with_agent(self, prompt, messages, request: gr.Request):
         """Handle agent interaction with proper session management."""
         # Get unique session ID for this user
-        session_id = get_user_session(request)
+        session_id = get_stable_session_id(request)
         session_data = get_session_data(session_id)
         
-        logger.info(f"Processing request for session {session_id[:8]}...")
+        logger.info(f"Processing request for session {session_id}...")
+        logger.info(f"Request client: {request.client.host if request and request.client else 'unknown'}")
+        logger.info(f"Request user-agent: {request.headers.get('user-agent', 'unknown')[:50] if request else 'unknown'}")
         
         # Check if we have a valid agent for this session
         if not session_data.get("agent"):
@@ -443,10 +467,12 @@ User Query: """
     def setup_api_key(self, api_key: str, max_steps: int, request: gr.Request) -> str:
         """Setup API key for the user's session."""
         # Get unique session ID for this user
-        session_id = get_user_session(request)
+        session_id = get_stable_session_id(request)
         session_data = get_session_data(session_id)
         
-        logger.info(f"Setting up API key for session {session_id[:8]}...")
+        logger.info(f"Setting up API key for session {session_id}...")
+        logger.info(f"Setup request client: {request.client.host if request and request.client else 'unknown'}")
+        logger.info(f"Setup request user-agent: {request.headers.get('user-agent', 'unknown')[:50] if request else 'unknown'}")
         
         # Check if API key is provided from interface
         if api_key and api_key.strip():