Update app.py

app.py

@@ -179,8 +179,10 @@ def to_markdown(file_path, end_pages, is_ocr, layout_mode, formula_enable, table
     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, "마크다운 읽는 중...")
@@ -194,13 +196,13 @@ def to_markdown(file_path, end_pages, is_ocr, layout_mode, formula_enable, table
     time.sleep(0.5)
 
     progress(100, "변환 완료!")
-    return md_content
+    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" 헤더로 구분되며, 
-    두 문서 모두 업로드된 경우 두 문서의 차이점 및 주요 특징을 비교 분석하는 추가 지시사항을 포함한다.
+    두 문서 모두 업로드된 경우 두 문서의 차이점, 장단점 및 주요 내용을 비교 분석하도록 추가 지시사항을 포함한다.
     """
     combined_md = ""
     if file_a is not None:
@@ -212,7 +214,7 @@ def to_markdown_comparison(file_a, file_b, end_pages, is_ocr, layout_mode, formu
         md_b = to_markdown(file_b, end_pages, is_ocr, layout_mode, formula_enable, table_enable, language, progress=progress)
         combined_md += md_b + "\n"
     if file_a is not None and file_b is not None:
-        combined_md += "### 비교 분석:\n두 문서의 차이점, 장단점 및 내용상의 주요 차이를 비교 분석하십시오.\n"
+        combined_md += "### 비교 분석:\n두 문서의 차이점, 장단점 및 주요 내용을 비교 분석하십시오.\n"
     return combined_md
 
 def init_model():
@@ -302,8 +304,7 @@ def convert_chat_messages_to_gradio_format(messages):
 
 def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
     """
-    Gemini 응답 스트리밍
-    (user_message가 공백이면 기본 문구로 대체)
+    Gemini 응답 스트리밍 (user_message가 공백이면 기본 문구로 대체)
     """
     if not user_message.strip():
         user_message = "...(No content from user)..."
@@ -318,7 +319,7 @@ def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
         response_buffer = ""
         thinking_complete = False
 
-        # "Thinking" 역할
+        # "Thinking" 역할 추가
         messages.append(
             ChatMessage(
                 role="assistant",
@@ -332,7 +333,6 @@ def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
             parts = chunk.candidates[0].content.parts
             current_chunk = parts[0].text
 
-            # 만약 parts가 2개라면, parts[0]는 thinking, parts[1]은 최종답변
             if len(parts) == 2 and not thinking_complete:
                 thought_buffer += current_chunk
                 messages[-1] = ChatMessage(
@@ -367,7 +367,7 @@ def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
 
 def user_message(msg: str, history: list, doc_text: str) -> tuple[str, list]:
     """
-    doc_text(마크다운) 사용해 질문 자동 변형
+    입력 프롬프트와 변환된 문서를 활용해 질문을 구성
     """
     if doc_text.strip():
         user_query = f"다음 문서를 참고하여 답변:\n\n{doc_text}\n\n질문: {msg}"
@@ -379,10 +379,7 @@ def user_message(msg: str, history: list, doc_text: str) -> tuple[str, list]:
 
 def reset_states(file_a, file_b):
     """
-    새 파일 업로드 시 
-    - chat_history -> 빈 리스트
-    - md_state -> 빈 문자열
-    - chatbot -> 빈 리스트
+    새 파일 업로드 시 chat_history와 md_state를 초기화 (Chatbot은 숨김 처리)
     """
     return [], "", []
 
@@ -403,20 +400,24 @@ if __name__ == "__main__":
             md_state = gr.State("")      # 변환된 마크다운 텍스트 (두 파일의 결과 통합)
             chat_history = gr.State([])  # ChatMessage 리스트
 
+            # 단일(숨김) Chatbot 컴포넌트: 화면에 보이지 않음
+            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)
                 convert_btn = gr.Button("비교용 변환하기")
 
+            # 파일 업로드 시 상태 초기화 (Chatbot은 숨김 처리)
             file_a.change(
                 fn=reset_states,
                 inputs=[file_a, file_b],
-                outputs=[chat_history, md_state, gr.Chatbot()]
+                outputs=[chat_history, md_state, hidden_chatbot]
             )
             file_b.change(
                 fn=reset_states,
                 inputs=[file_a, file_b],
-                outputs=[chat_history, md_state, gr.Chatbot()]
+                outputs=[chat_history, md_state, hidden_chatbot]
             )
 
             max_pages = gr.Slider(1, 20, 10, visible=False)
@@ -446,6 +447,7 @@ if __name__ == "__main__":
                 chat_input = gr.Textbox(lines=1, placeholder="질문을 입력하세요...")
                 clear_btn = gr.Button("대화 초기화")
 
+            # 채팅 입력 후, LLM 응답은 hidden_chatbot에 저장(화면에 보이지 않음)
             chat_input.submit(
                 fn=user_message,
                 inputs=[chat_input, chat_history, md_state],
@@ -453,7 +455,7 @@ if __name__ == "__main__":
             ).then(
                 fn=stream_gemini_response,
                 inputs=[chat_input, chat_history],
-                outputs=gr.Chatbot()
+                outputs=hidden_chatbot
             )
 
             def clear_all():
@@ -462,7 +464,7 @@ if __name__ == "__main__":
             clear_btn.click(
                 fn=clear_all,
                 inputs=[],
-                outputs=[chat_history, md_state, gr.Chatbot()]
+                outputs=[chat_history, md_state, hidden_chatbot]
             )
 
     demo.launch(server_name="0.0.0.0", server_port=7860, debug=True, ssr_mode=True)