Update app.py

app.py

@@ -7,6 +7,9 @@ import os
 from huggingface_hub import InferenceClient
 import json
 from langchain_community.agent_toolkits import GmailToolkit
+import time
+import requests
+from datetime import datetime
 
 
 # Initialize FastAPI app
@@ -18,7 +21,10 @@ def read_root():
     return {"message": "Connection"}
 
 
-
+@app.get("/ping")
+def ping():
+    return {"status": "alive"}
+    
 UPLOAD_FOLDER = "uploaded_pdfs"
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
@@ -37,6 +43,29 @@ async def upload_pdf(file: UploadFile = File(...)):
 def list_uploaded_files():
     files = os.listdir(UPLOAD_FOLDER)
     return {"files": files}
+
+
+
+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)
+
+# Call the function
+keep_alive()
+
+
+
 # # API endpoint for prediction
 # @app.post("/predict")
 # async def predict_image(file: UploadFile = File(...)):