Update app.py

app.py

@@ -3,6 +3,18 @@ import requests
 
 API_URL = "https://rahul7star-FramePack-F1-DiffusionForce.hf.space/api/generate/"
 
+HEALTH_API_URL = "https://rahul7star-FramePack-F1-DiffusionForce.hf.space/api/healthcheck"
+
+def check_health():
+    try:
+        response = requests.get(HEALTH_API_URL)
+        if response.status_code == 200:
+            return f"✅ API is healthy: {response.json()}"
+        else:
+            return f"❌ API Error: {response.status_code} - {response.text}"
+    except Exception as e:
+        return f"❌ Exception occurred: {str(e)}"
+
 def call_framepack_api(
     input_image,
     prompt,
@@ -92,6 +104,10 @@ with gr.Blocks() as demo:
             fp8_optimization = gr.Checkbox(label="FP8 Optimization", value=False)
 
             generate_btn = gr.Button("Generate")
+            health_btn = gr.Button("Check API Health")
+            health_output = gr.Textbox(label="Health Check Result")
+
+            health_btn.click(fn=check_health, inputs=[], outputs=[health_output])
 
         with gr.Column():
             video_output = gr.Video(label="Generated Video", autoplay=True)