Create app.py

app.py

@@ -0,0 +1,81 @@
+import os
+import gradio as gr
+import requests
+from langchain_community.document_loaders import UnstructuredPDFLoader
+import json
+
+# 環境変数からDIFY_BASE_URLとDIFY_API_KEY_MYWORKFLOWを取得
+DIFY_BASE_URL = os.environ.get("DIFY_BASE_URL", "")
+DIFY_API_KEY_MYWORKFLOW = os.environ.get("DIFY_API_KEY_MYWORKFLOW", "")
+
+def run_workflow(file):
+    try:
+        if not file:
+            return "PDFファイルを選択してください。", ""
+
+        loader = UnstructuredPDFLoader(file)
+        data = loader.load()
+        raw_text = data[0].page_content
+
+        inputs = {
+            "url": "",
+            "knowledge": raw_text
+        }
+        yield raw_text,"loading...",{}
+
+        url = DIFY_BASE_URL + "/workflows/run"
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {DIFY_API_KEY_MYWORKFLOW}"
+        }
+        data = {
+            "inputs": inputs,
+            "query": "",
+            "response_mode": "streaming",
+            "user": "abc_123",
+        }
+        response = requests.post(url, headers=headers, json=data, stream=True)
+        response.raise_for_status()
+
+        assistant_message = ""
+        outputs = {}
+
+        for chunk in response.iter_lines(delimiter=b"\n\n"):
+            if chunk:
+                chunk_data = chunk.decode("utf-8").strip()
+                if chunk_data.startswith("data:"):
+                    json_data = chunk_data[6:]  # "data: "を取り除く
+                    if json_data:
+                        result = json.loads(json_data)
+                        if result.get("event") == "text_chunk":
+                            answer = result.get("data", "").get("text", "")
+                            assistant_message += str(answer)
+                            yield raw_text,assistant_message,result.get("data", "")
+                        elif result.get("event") == "workflow_finished":
+                            outputs = result.get('data', "")
+                            yield raw_text,assistant_message,outputs
+
+        # print(outputs.get('result', ""))
+        # print(outputs.get('row_content', ""))
+        # print(outputs.get('url', ""))
+
+        # return raw_text, assistant_message, outputs
+    except Exception as e:
+        error_message = str(e)
+        print(f"Error: {error_message}")
+        return "error", error_message, {}
+
+iface = gr.Interface(
+    fn=run_workflow,
+    inputs=[gr.File(label="PDFファイルをアップロード", file_types=[".pdf"])],
+    outputs=[
+        gr.Textbox(label="生テキスト", show_copy_button=True, max_lines=5),
+        gr.Markdown(),
+        gr.JSON()
+    ],
+    title="PDF to Dify Workflow",
+    description="PDFファイルを入力すると、Dify APIのワークフローによって処理された結果が表示されます。"
+)
+
+if __name__ == "__main__":
+    iface.queue().launch()