Update app.py

app.py

@@ -330,24 +330,7 @@ def clear_agent_only(session_id: str):
                 del user_sessions[session_id]["agent"]
             logger.info(f"Session {session_id[:8]}... agent cleared")
 
-def cleanup_old_sessions():
-    """Clean up sessions older than 1 hour."""
-    with session_lock:
-        current_time = datetime.now()
-        sessions_to_remove = []
-        
-        for session_id, session_data in user_sessions.items():
-            if session_data.get("created_at"):
-                time_diff = current_time - session_data["created_at"]
-                if time_diff.total_seconds() > 3600:  # 1 hour
-                    sessions_to_remove.append(session_id)
-        
-        for session_id in sessions_to_remove:
-            del user_sessions[session_id]
-            logger.info(f"Removed old session {session_id[:8]}...")
-        
-        if sessions_to_remove:
-            logger.info(f"Cleaned up {len(sessions_to_remove)} old sessions")
+
 
 class GradioUI:
     """A one-line interface to launch your agent in Gradio"""
@@ -772,7 +755,7 @@ This AI agent specializes in automated vulnerability research and analysis, buil
                             # API Key Configuration Section
                             with gr.Accordion("🔑 API Configuration", open=False):
                                 gr.Markdown("**Configure your Hugging Face API Key**")
-                                gr.Markdown("🔒 **Security**: Each user has an independent session. Your API key is stored securely and automatically cleared after 1 hour of inactivity.")
+                                gr.Markdown("🔒 **Security**: Your API key is only kept during this session.")
                                 gr.Markdown("Get your API key from: https://huggingface.co/settings/tokens")
                                 
                                 api_key_input = gr.Textbox(
@@ -920,7 +903,7 @@ This AI agent specializes in automated vulnerability research and analysis, buil
                         # API Key Configuration Section for Mobile
                         with gr.Accordion("🔑 API Configuration", open=False):
                             gr.Markdown("**Configure your Hugging Face API Key**")
-                            gr.Markdown("🔒 **Security**: Each user has an independent session. Your API key is stored securely and automatically cleared after 1 hour of inactivity.")
+                            gr.Markdown("🔒 **Security**: Your API key is only kept during this session.")
                             gr.Markdown("Get your API key from: https://huggingface.co/settings/tokens")
                             
                             mobile_api_key_input = gr.Textbox(
@@ -1081,19 +1064,7 @@ This AI agent specializes in automated vulnerability research and analysis, buil
                             [text_input, launch_research_btn],
                         )
 
-        # Start periodic cleanup of old sessions
-        def periodic_cleanup():
-            import time
-            while True:
-                try:
-                    cleanup_old_sessions()
-                    time.sleep(300)  # Run every 5 minutes
-                except Exception as e:
-                    logger.error(f"Error in periodic cleanup: {e}")
-                    time.sleep(300)
-        
-        cleanup_thread = threading.Thread(target=periodic_cleanup, daemon=True)
-        cleanup_thread.start()
+
         
         demo.launch(debug=True, **kwargs)