server monitor

app.py

@@ -7,6 +7,7 @@ from gradio import State
 import asyncio
 import json
 import asyncio
+import threading
 
 # Function to start the ochat server
 @spaces.GPU
@@ -26,9 +27,28 @@ def start_ochat_server():
     except Exception as e:
         return f"Failed to start ochat server: {e}"
 
+async def monitor_server():
+    while True:
+        async with aiohttp.ClientSession() as session:
+            try:
+                async with session.get("http://localhost:18888/docs") as response:
+                    if response.status == 200:
+                        print("Server is running.")
+                    else:
+                        print("Server is not running. Attempting to restart...")
+                        start_ochat_server()
+            except aiohttp.ClientError:
+                print("Server is not running. Attempting to restart...")
+                start_ochat_server()
+
+        await asyncio.sleep(60)  # Check every 60 seconds
 
 start_ochat_server()
 
+# Start the monitoring in a separate thread
+thread = threading.Thread(target=monitor_server)
+thread.start()
+
 # Function to send a message to the ochat server and get a response
 async def chat_with_ochat(message):
     base_url = "http://localhost:18888"