Create app.py

app.py

@@ -0,0 +1,147 @@
+import gradio as gr
+import requests
+import json
+import time
+import os
+
+# --- 配置資訊 ---
+API_KEY = "46b4a365e01c454ebcfbe41537af23d4"
+WORKFLOW_ID = "958842411961233410"
+
+# RunningHub API 端點
+UPLOAD_URL = "https://www.runninghub.cn/api/resource/openapi/upload"
+CREATE_TASK_URL = "https://www.runninghub.cn/task/openapi/create"
+GET_TASK_INFO_URL = "https://www.runninghub.cn/task/openapi/getTaskInfo"
+
+# 你的工作流中需要修改的節點ID
+# 節點 141 用來載入圖片，我們需要修改它的 'image' 欄位
+LOAD_IMAGE_NODE_ID = "141"
+POSITIVE_PROMPT_NODE_ID = "144" # 正向提示詞
+SEED_NODE_ID = "132" # 種子值
+
+# --- 核心邏輯函式 ---
+# (此部分與您之前收到的程式碼相同，無需修改)
+
+def process_image_with_runninghub(uploaded_image, positive_prompt, seed):
+    """
+    處理上傳圖片，調用 RunningHub API 進行圖像生成。
+    """
+    if uploaded_image is None:
+        return "請先上傳一張圖片！", None
+    
+    # 步驟 1: 上傳圖片到 RunningHub
+    yield "正在上傳圖片...", None
+    try:
+        files = {'file': open(uploaded_image, 'rb')}
+        upload_response = requests.post(
+            UPLOAD_URL,
+            headers={"apiKey": API_KEY},
+            files=files
+        )
+        upload_response.raise_for_status()
+        upload_data = upload_response.json()
+        
+        if upload_data.get("code") != 0:
+            return f"圖片上傳失敗！訊息: {upload_data.get('msg')}", None
+        
+        rh_image_path = upload_data["data"]["filePath"]
+        yield f"圖片上傳成功！路徑: {rh_image_path}", None
+
+    except requests.exceptions.RequestException as e:
+        return f"上傳連線錯誤: {e}", None
+    
+    # 步驟 2: 建立 ComfyUI 任務
+    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
+    }
+
+    headers = {'Content-Type': 'application/json'}
+    
+    try:
+        create_response = requests.post(CREATE_TASK_URL, headers=headers, 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('code')}, 訊息: {create_data.get('msg')}", None
+        
+        task_id = create_data["data"]["taskId"]
+        yield f"任務已建立！任務ID: {task_id}", None
+
+    except requests.exceptions.RequestException as e:
+        return f"任務建立連線錯誤: {e}", None
+
+    # 步驟 3: 輪詢任務狀態
+    max_retries = 30
+    retry_interval = 5 # 5 秒
+
+    for i in range(max_retries):
+        yield f"正在查詢任務狀態... (第 {i+1} 次)", None
+        time.sleep(retry_interval)
+        
+        status_payload = {"apiKey": API_KEY, "taskId": task_id}
+        
+        try:
+            status_response = requests.post(GET_TASK_INFO_URL, headers=headers, 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 "查詢任務超時，請稍後使用任務ID手動查詢。", None
+    return "查詢任務超時，請稍後使用任務ID手動查詢。", None
+
+# --- Gradio 介面設定 ---
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# RunningHub - 圖像處理工作流")
+    gr.Markdown("上傳一張圖片，輸入提示詞和種子值，然後啟動遠端圖像處理任務。")
+    
+    with gr.Row():
+        with gr.Column():
+            with gr.Row():
+                image_input = gr.Image(type="filepath", label="上傳圖片")
+            
+            with gr.Column():
+                prompt_input = gr.Textbox(lines=3, label="正向提示詞 (Positive Prompt)", placeholder="輸入你想要生成的內容...")
+                seed_input = gr.Number(label="種子值 (Seed)", value=-1, precision=0)
+            
+            process_button = gr.Button("開始處理圖像", variant="primary")
+            
+        with gr.Column():
+            with gr.Row():
+                image_output = gr.Image(label="生成的圖像")
+            with gr.Row():
+                status_output = gr.Textbox(label="任務狀態", interactive=False)
+
+    process_button.click(
+        fn=process_image_with_runninghub,
+        inputs=[image_input, prompt_input, seed_input],
+        outputs=[status_output, image_output]
+    )
+
+if __name__ == "__main__":
+    demo.launch()