Update app.py

app.py

@@ -145,7 +145,11 @@ def replace_image_with_base64(markdown_text, image_dir_path):
 def to_pdf(file_path):
     """
     이미지(JPG/PNG 등)를 PDF로 컨버팅.
+    TXT, CSV 파일인 경우 변환 없이 원본 경로를 반환한다.
     """
+    ext = Path(file_path).suffix.lower()
+    if ext in ['.txt', '.csv']:
+        return file_path
     with pymupdf.open(file_path) as f:
         if f.is_pdf:
             return file_path
@@ -159,50 +163,54 @@ def to_pdf(file_path):
 
 def to_markdown(file_path, end_pages, is_ocr, layout_mode, formula_enable, table_enable, language, progress=gr.Progress(track_tqdm=False)):
     """
-    업로드된 PDF/이미지 -> PDF 변환 -> 마크다운 변환
+    업로드된 PDF/이미지 또는 TXT/CSV -> 마크다운 변환
     (프로그레스 바 표시용)
     """
-    progress(0, "PDF로 변환 중...")
-    file_path = to_pdf(file_path)
-    time.sleep(0.5)
-
-    if end_pages > 20:
-        end_pages = 20
-
-    progress(20, "문서 파싱 중...")
-    local_md_dir, file_name = parse_pdf(file_path, './output', end_pages - 1, is_ocr,
-                                        layout_mode, formula_enable, table_enable, language)
-    time.sleep(0.5)
-
-    progress(50, "압축(zip) 생성 중...")
-    archive_zip_path = os.path.join("./output", compute_sha256(local_md_dir) + ".zip")
-    zip_archive_success = compress_directory_to_zip(local_md_dir, archive_zip_path)
-    if zip_archive_success == 0:
-        logger.info("압축 성공")
-        status_message = "\n\n**변환 완료 (압축 성공)**"
+    ext = Path(file_path).suffix.lower()
+    if ext in ['.txt', '.csv']:
+        progress(0, "파일 읽는 중...")
+        with open(file_path, 'r', encoding='utf-8') as f:
+            txt_content = f.read()
+        time.sleep(0.5)
+        progress(50, "파일 내용 처리 중...")
+        progress(100, "변환 완료!")
+        return f"```{txt_content}```\n\n**변환 완료 (텍스트/CSV 파일)**"
     else:
-        logger.error("압축 실패")
-        status_message = "\n\n**변환 완료 (압축 실패)**"
-    time.sleep(0.5)
-
-    progress(70, "마크다운 읽는 중...")
-    md_path = os.path.join(local_md_dir, file_name + ".md")
-    with open(md_path, 'r', encoding='utf-8') as f:
-        txt_content = f.read()
-    time.sleep(0.5)
-
-    progress(90, "이미지 base64 변환 중...")
-    md_content = replace_image_with_base64(txt_content, local_md_dir)
-    time.sleep(0.5)
-
-    progress(100, "변환 완료!")
-    return md_content + status_message
+        progress(0, "PDF로 변환 중...")
+        file_path = to_pdf(file_path)
+        time.sleep(0.5)
+        if end_pages > 20:
+            end_pages = 20
+        progress(20, "문서 파싱 중...")
+        local_md_dir, file_name = parse_pdf(file_path, './output', end_pages - 1, is_ocr,
+                                            layout_mode, formula_enable, table_enable, language)
+        time.sleep(0.5)
+        progress(50, "압축(zip) 생성 중...")
+        archive_zip_path = os.path.join("./output", compute_sha256(local_md_dir) + ".zip")
+        zip_archive_success = compress_directory_to_zip(local_md_dir, archive_zip_path)
+        if zip_archive_success == 0:
+            logger.info("압축 성공")
+            status_message = "\n\n**변환 완료 (압축 성공)**"
+        else:
+            logger.error("압축 실패")
+            status_message = "\n\n**변환 완료 (압축 실패)**"
+        time.sleep(0.5)
+        progress(70, "마크다운 읽는 중...")
+        md_path = os.path.join(local_md_dir, file_name + ".md")
+        with open(md_path, 'r', encoding='utf-8') as f:
+            txt_content = f.read()
+        time.sleep(0.5)
+        progress(90, "이미지 base64 변환 중...")
+        md_content = replace_image_with_base64(txt_content, local_md_dir)
+        time.sleep(0.5)
+        progress(100, "변환 완료!")
+        return md_content + status_message
 
 def to_markdown_comparison(file_a, file_b, end_pages, is_ocr, layout_mode, formula_enable, table_enable, language, progress=gr.Progress(track_tqdm=False)):
     """
-    두 개의 파일을 변환하여 A/B 비교를 위한 마크다운을 생성한다.
-    각 파일의 변환 결과는 "문서 A", "문서 B" 헤더로 구분되며, 
-    두 문서 모두 업로드된 경우 두 문서의 차이점, 장단점 및 주요 내용을 비교 분석하도록 추가 지시사항을 포함한다.
+    두 개의 파일을 변환하여 A/B 비교용 마크다운 생성.
+    각 파일은 "문서 A", "문서 B" 헤더로 구분되며,
+    두 파일 모두 업로드된 경우 추가로 비교 분석 지시사항을 포함한다.
     """
     combined_md = ""
     if file_a is not None:
@@ -392,7 +400,7 @@ if __name__ == "__main__":
             gr.HTML("""
             <div class="title-area">
                 <h1>VisionOCR</h1>
-                <p>두 개의 PDF/이미지 파일을 업로드하여 A/B 비교 후, 추론 LLM과 대화합니다.<br>
+                <p>두 개의 PDF/이미지/텍스트/CSV 파일을 업로드하여 A/B 비교 후, 추론 LLM과 대화합니다.<br>
                 한 파일만 업로드하면 해당 파일로 분석합니다.</p>
             </div>
             """)
@@ -404,8 +412,8 @@ if __name__ == "__main__":
             hidden_chatbot = gr.Chatbot(visible=False)
 
             with gr.Row():
-                file_a = gr.File(label="문서 A 업로드", file_types=[".pdf", ".png", ".jpeg", ".jpg"], interactive=True)
-                file_b = gr.File(label="문서 B 업로드", file_types=[".pdf", ".png", ".jpeg", ".jpg"], interactive=True)
+                file_a = gr.File(label="문서 A 업로드", file_types=[".pdf", ".png", ".jpeg", ".jpg", ".txt", ".csv"], interactive=True)
+                file_b = gr.File(label="문서 B 업로드", file_types=[".pdf", ".png", ".jpeg", ".jpg", ".txt", ".csv"], interactive=True)
                 convert_btn = gr.Button("비교용 변환하기")
 
             # 파일 업로드 시 상태 초기화 (Chatbot은 숨김 처리)
@@ -447,7 +455,7 @@ if __name__ == "__main__":
                 chat_input = gr.Textbox(lines=1, placeholder="질문을 입력하세요...")
                 clear_btn = gr.Button("대화 초기화")
 
-            # 채팅 입력 후, LLM 응답은 hidden_chatbot에 저장(화면에 보이지 않음)
+            # 채팅 입력 후 LLM 응답은 hidden_chatbot에 저장(화면에 보이지 않음)
             chat_input.submit(
                 fn=user_message,
                 inputs=[chat_input, chat_history, md_state],