Update app.py

app.py

@@ -1,47 +1,118 @@
 import gradio as gr
 import requests
+import json
+import time
 
 # --- 配置資訊 ---
 API_KEY = "d72ef10876b9403a9479c6e31f27a4be".strip()
+WORKFLOW_ID = "1958842411961233410"  # 你的工作流ID
 UPLOAD_URL = "https://www.runninghub.cn/task/openapi/upload"
+CREATE_TASK_URL = "https://www.runninghub.cn/task/openapi/create"
+GET_TASK_INFO_URL = "https://www.runninghub.cn/task/openapi/getTaskInfo"
 HEADERS = {"Host": "www.runninghub.cn"}
 
+# 你的工作流節點 ID
+LOAD_IMAGE_NODE_ID = "141"
+POSITIVE_PROMPT_NODE_ID = "144"
+SEED_NODE_ID = "132"
+
 # --- 核心函式 ---
-def test_upload_image(image_path):
+def process_image_workflow(image_path, positive_prompt, seed=-1):
+    """
+    1. 上傳圖片
+    2. 建立 RunningHub 工作流任務
+    3. 輪詢任務狀態
+    4. 返回生成圖片
+    """
     if image_path is None:
-        return "請先上傳圖片！"
+        return "請先上傳圖片！", None
 
+    # --- 步驟 1: 上傳圖片 ---
+    yield "正在上傳圖片...", None
     try:
         with open(image_path, "rb") as f:
             files = {"file": f}
-            data = {
-                "apiKey": API_KEY,
-                "fileType": "image"  # 圖片固定為 image
-            }
+            data = {"apiKey": API_KEY, "fileType": "image"}
+            upload_response = requests.post(UPLOAD_URL, headers=HEADERS, data=data, files=files)
+            upload_response.raise_for_status()
+            upload_data = upload_response.json()
+            print(upload_data)  # 除錯
+
+            if upload_data.get("code") != 0:
+                return f"圖片上傳失敗！訊息: {upload_data.get('msg')}", None
 
-            response = requests.post(UPLOAD_URL, headers=HEADERS, data=data, files=files)
-            response.raise_for_status()
-            result = response.json()
-            print(result)  # 除錯用
+            rh_image_path = upload_data["data"]["fileName"]
+            yield f"圖片上傳成功！fileName: {rh_image_path}", None
 
-            if result.get("code") == 0:
-                return f"上傳成功！fileName: {result['data']['fileName']}"
-            else:
-                return f"上傳失敗！訊息: {result.get('msg')}"
+    except requests.exceptions.RequestException as e:
+        return f"上傳連線錯誤: {e}", None
 
+    # --- 步驟 2: 建立工作流任務 ---
+    yield "正在建立任務...", None
+    node_info_list = [
+        {"nodeId": LOAD_IMAGE_NODE_ID, "fieldName": "image", "fieldValue": rh_image_path},
+        {"nodeId": POSITIVE_PROMPT_NODE_ID, "fieldName": "ShowText_0", "fieldValue": positive_prompt},
+        {"nodeId": SEED_NODE_ID, "fieldName": "seed", "fieldValue": seed}
+    ]
+    payload = {"apiKey": API_KEY, "workflowId": WORKFLOW_ID, "nodeInfoList": node_info_list}
+    try:
+        create_response = requests.post(CREATE_TASK_URL, headers={"Content-Type": "application/json"}, data=json.dumps(payload))
+        create_response.raise_for_status()
+        create_data = create_response.json()
+        if create_data.get("code") != 0:
+            return f"任務建立失敗！訊息: {create_data.get('msg')}", None
+        task_id = create_data["data"]["taskId"]
+        yield f"任務建立成功！TaskID: {task_id}", None
     except requests.exceptions.RequestException as e:
-        return f"連線錯誤: {e}"
+        return f"任務建立連線錯誤: {e}", None
+
+    # --- 步驟 3: 輪詢任務狀態 ---
+    max_retries = 30
+    retry_interval = 5
+    for i in range(max_retries):
+        yield f"查詢任務狀態... (第 {i+1} 次)", None
+        time.sleep(retry_interval)
+        try:
+            status_payload = {"apiKey": API_KEY, "taskId": task_id}
+            status_response = requests.post(GET_TASK_INFO_URL, headers={"Content-Type": "application/json"}, data=json.dumps(status_payload))
+            status_response.raise_for_status()
+            status_data = status_response.json()
+            current_status = status_data["data"]["taskStatus"]
+            if current_status == "SUCCESS":
+                image_url = status_data["data"]["imageInfoList"][0]["fileUrl"]
+                yield f"任務完成！", image_url
+                return f"任務完成！", image_url
+            elif current_status == "FAILED":
+                error_msg = status_data.get("data", {}).get("errorMsg", "Unknown error")
+                yield f"任務失敗！錯誤訊息: {error_msg}", None
+                return f"任務失敗！錯誤訊息: {error_msg}", None
+        except requests.exceptions.RequestException as e:
+            yield f"查詢連線錯誤: {e}", None
+            return f"查詢連線錯誤: {e}", None
+
+    yield "任務查詢超時，請使用 TaskID 手動查詢。", None
+    return "任務查詢超時，請使用 TaskID 手動查詢。", None
 
 # --- Gradio 介面 ---
 with gr.Blocks() as demo:
-    gr.Markdown("# RunningHub 上傳測試")
-    gr.Markdown("上傳一張圖片，測試 API Key 是否有效。")
+    gr.Markdown("# RunningHub 工作流圖生圖")
+    gr.Markdown("上傳圖片，輸入正向提示詞與種子值，啟動遠端工作流任務。")
 
-    image_input = gr.Image(type="filepath", label="上傳圖片")
-    status_output = gr.Textbox(label="狀態", interactive=False)
-    upload_button = gr.Button("測試上傳")
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(type="filepath", label="上傳圖片")
+            prompt_input = gr.Textbox(lines=3, label="正向提示詞 (Positive Prompt)", placeholder="輸入你想生成的內容...")
+            seed_input = gr.Number(label="種子值 (Seed)", value=-1, precision=0)
+            process_button = gr.Button("開始處理圖像")
+        with gr.Column():
+            status_output = gr.Textbox(label="任務狀態", interactive=False)
+            image_output = gr.Image(label="生成圖片")
 
-    upload_button.click(fn=test_upload_image, inputs=image_input, outputs=status_output)
+    process_button.click(
+        fn=process_image_workflow,
+        inputs=[image_input, prompt_input, seed_input],
+        outputs=[status_output, image_output]
+    )
 
 if __name__ == "__main__":
     demo.launch(share=True)