From 979b9d611e3571bd10e0070ce59f7453d74405c2 Mon Sep 17 00:00:00 2001
From: ChungHwan Han <51526347+hanchchch@users.noreply.github.com>
Date: Sun, 9 Apr 2023 04:31:38 +0900
Subject: [PATCH] fix: remove file:// of filepath when referring it (#38)

---
 api/main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/api/main.py b/api/main.py
index 898207b..1c9cd7e 100644
--- a/api/main.py
+++ b/api/main.py
@@ -68,7 +68,7 @@ async def execute(request: ExecuteRequest) -> ExecuteResponse:
         return {"answer": str(e), "files": []}
 
     files = re.findall(r"\[file://\S*\]", res["output"])
-    files = [file[1:-1] for file in files]
+    files = [file[1:-1].split("file://")[1] for file in files]
 
     return {
         "answer": res["output"],
@@ -97,7 +97,7 @@ async def execute_async(execution_id: str):
     if execution.status == "SUCCESS" and execution.result:
         output = execution.result.get("output", "")
         files = re.findall(r"\[file://\S*\]", output)
-        files = [file[1:-1] for file in files]
+        files = [file[1:-1].split("file://")[1] for file in files]
         result = {
             "answer": output,
             "files": [uploader.upload(file) for file in files],