Update app.py

app.py

@@ -95,10 +95,13 @@ def extract_info(image_path):
     # 7. CSV 파일로 저장 (임시 디렉토리에 저장하고 파일 경로 반환)
     tmp_dir = tempfile.mkdtemp()
     csv_path = os.path.join(tmp_dir, "namecard_example.csv")
-    df_result.to_csv(csv_path, index=False)
+
+    csv_io = io.BytesIO()
+    df_result.to_csv(csv_io, index=False)
+    csv_io.seek(0)
 
     # JSON 문자열, 테이블 데이터프레임, CSV 파일 경로 반환
-    return json_result, df_result, csv_path
+    return json_result, df_result, csv_io 
 
 # 🔧 Gradio 인터페이스 구성
 with gr.Blocks() as demo:
@@ -125,18 +128,15 @@ with gr.Blocks() as demo:
         outputs=[json_output, table_output, file_output]
     )
 
-    # 다운로드 트리거 함수: 경로를 파일 객체로 열어 반환
-    def open_csv_for_download(file_path):
-        return open(file_path, "rb")
-    
-    # 다운로드 버튼 클릭 시 실제 다운로드 실행
+
+    csv_output = gr.File(label="", visible=False, file_types=[".csv"])
+
     download_btn.click(
-        fn=open_csv_for_download,
-        inputs=file_output,
-        outputs=file_output
+        fn=lambda file: file,
+        inputs=csv_output,
+        outputs=csv_output
     )
 
-
 # 앱 실행
 if __name__ == "__main__":
     demo.launch()