Update app.py

app.py

@@ -22,22 +22,42 @@ import json
 import shutil
 import os
 from fastapi import FastAPI, File, UploadFile
+import time
+import requests
+from datetime import datetime
+
 
 app = FastAPI()
 
 @app.get("/")
 def read_root():
     app.state.vector_index = None
+    keep_alive()
     return {"message": "Connected"}
 
+def keep_alive(space_url="https://1mr-apigmail.hf.space/ping", interval_hours=5):
+    while True:
+        try:
+            print(f"🔄 Pinging {space_url} at {datetime.now()}")
+            response = requests.get(space_url)
+
+            if response.status_code == 200:
+                print("")
+            else:
+                print("")
+        except Exception as e:
+            print("")
+        
+        time.sleep(interval_hours * 3600)
+
+# keep_alive()
+
 @tool
 def retrieve(query_text):
     """
     Retrieves relevant information from a vector index based on a query.
-
     Parameters:
     - query_text (str): Query to search for relevant information.
-
     Returns:
     - str: Retrieved text from the document.
     """
@@ -84,7 +104,6 @@ async def upload_file(file: UploadFile = File(...)):
     return {"message": message, "filename": file.filename}
 
 
-
 class QueryRequest(BaseModel):
     message: str