Update codriao_web_cli.py

codriao_web_cli.py

@@ -143,8 +143,16 @@ if __name__ == "__main__":
     try:
         mode = input("Start Codriao in [web] or [cli]? ").strip().lower()
         if mode == "cli":
-            guardian_cli()
+            launch_guardian()
         else:
-            demo.launch()
+            demo.launch()  # This will launch Gradio if mode is 'web'
+
+    finally:
+        asyncio.run(ai_core.http_session.close())  # Ensures the HTTP session is properly closed
+
+# Function to launch the Codriao Guardian CLI and ensure a clean shutdown
+def launch_guardian():
+    try:
+        guardian_cli()  # This is where the Guardian CLI is launched
     finally:
-        asyncio.run(ai_core.http_session.close())
+        asyncio.run(ai_core.shutdown())