Update app.py

app.py

@@ -37,10 +37,19 @@ BACKEND_URL = "LiuZichen/MagicQuillV2"
 SAM_URL = "LiuZichen/MagicQuillHelper"
 
 print(f"Connecting to backend at: {BACKEND_URL}")
-backend_client = Client(BACKEND_URL, hf_token=hf_token)
+
+try:
+    backend_client = Client(BACKEND_URL, hf_token=hf_token)
+except Exception as e:
+    print(f"Failed to connect to backend: {e}")
+    backend_client = None
 
 print(f"Connecting to SAM client at: {SAM_URL}")
-sam_client = Client(SAM_URL, hf_token=hf_token)
+try:
+    sam_client = Client(SAM_URL, hf_token=hf_token)
+except Exception as e:
+    print(f"Failed to connect to SAM client: {e}")
+    sam_client = None
 
 # --- Helper Functions ---
 
@@ -93,6 +102,7 @@ def generate_image_handler(x, negative_prompt, fine_edge, fix_perspective, grow_
     return x
 
 # --- Gradio UI ---
+
 with gr.Blocks(title="MagicQuill V2") as demo:
     with gr.Row():
         ms = MagicQuillV2()
@@ -303,9 +313,5 @@ async def segmentation(request: Request):
         print(f"Error in segmentation: {e}")
         return {"error": str(e)}
 
-# Mount the Gradio app
-app = gr.mount_gradio_app(app, demo, path="/")
-
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
-
+    uvicorn.run(app, host="0.0.0.0", port=7860)