Update backend/app.py

backend/app.py

@@ -6,8 +6,19 @@ from fastapi.responses import JSONResponse
 from chainlit.auth import create_jwt
 from chainlit.server import app
 import chainlit as cl
-
-
+import requests
+
+def connect_to_ollama():
+    try:
+        response = requests.get('http://localhost:11434')
+        if response.status_code == 200:
+            print('Connected to ollama server')
+        else:
+            print('Failed to connect to ollama server')
+    except requests.exceptions.RequestException as e:
+        print(f'Error: could not connect to ollama server - {str(e)}')
+        
+connect_to_ollama()
 
 @app.get("/custom-auth")
 async def custom_auth():