Update app.py

app.py

@@ -1,24 +1,56 @@
-import gradio as gr
-from OCR_test import process_image  
-
-def detect_objects(image):
-    try:
-        # Call the `process_image` function with the image path
-        result = process_image(image)
-        return result
-    except Exception as e:
-        # Return any errors encountered during processing
-        return f"Error: {e}"
-
-# Define the Gradio interface
-interface = gr.Interface(
-    fn=detect_objects,
-    inputs=gr.Image(type="filepath", label="Upload Image"),  # Set to "filepath" to pass file path directly
-    outputs=gr.Textbox(label="Detection Results"),
-    title="OCR Detection",
-    description="Upload an image to get OCR results."
-)
-
-# Launch the Gradio app
-if __name__ == "__main__":
+import gradio as gr
+import threading
+import time
+import requests
+from OCR_test import process_image
+
+# ====== Keep-Alive Background Thread ======
+def keep_alive_worker():
+    # 替换为你的 HF Space 实际访问 URL（注意格式！）
+    space_url = "https://fcu52005505--test.hf.space"
+    interval_sec = 60  # 1 分钟 = 60 秒
+    
+    print(f"[Keep-Alive] Starting background pinger. Pinging {space_url} every {interval_sec} seconds...")
+    
+    while True:
+        try:
+            # 发送 HEAD 请求（轻量，不下载内容）
+            response = requests.head(space_url, timeout=10)
+            status = response.status_code
+            print(f"[Keep-Alive] Pinged at {time.strftime('%Y-%m-%d %H:%M:%S')} | Status: {status}")
+        except Exception as e:
+            print(f"[Keep-Alive] ⚠️ Ping failed at {time.strftime('%Y-%m-%d %H:%M:%S')}: {e}")
+        
+        time.sleep(interval_sec)
+
+# 启动后台线程（daemon=True 保证主线程退出时它也退出）
+threading.Thread(target=keep_alive_worker, daemon=True).start()
+
+# ====== OCR Inference Function ======
+def detect_objects(image):
+    try:
+        if image is None:
+            return "Error: No image uploaded."
+        # Call the `process_image` function with the image path
+        result = process_image(image)
+        return result
+    except Exception as e:
+        return f"Error: {e}"
+
+# ====== Gradio Interface ======
+interface = gr.Interface(
+    fn=detect_objects,
+    inputs=gr.Image(type="filepath", label="Upload Image"),
+    outputs=gr.Textbox(label="Detection Results"),
+    title="OCR Detection (Auto-Keep-Alive Enabled)",
+    description="Upload an image to get OCR results. This Space pings itself every 60s to avoid sleeping.",
+    examples=[
+        # 可选：添加示例图片路径（若你有放在 Space repo 的 example_images/）
+        # ["example_images/sample1.jpg"]
+    ]
+)
+
+# ====== Launch ======
+if __name__ == "__main__":
+    # 注意：HF Spaces 会忽略 `server_name` 和 `server_port`
     interface.launch()