Update app.py

app.py

@@ -882,9 +882,31 @@ def convert_hwp_to_markdown(input_path: str) -> tuple:
         return text, None
     return None, error
 
+def convert_to_odt_subprocess(input_path, output_dir):
+    output_path = os.path.join(output_dir, "output.odt")
+    try:
+        result = subprocess.run(['hwp5odt', '--output', output_path, input_path], capture_output=True, timeout=120)
+        if result.returncode == 0 and os.path.exists(output_path):
+            return output_path, None
+    except:
+        pass
+    return None, "ODT 변환 실패"
+
+def convert_to_xml_subprocess(input_path, output_dir):
+    output_path = os.path.join(output_dir, "output.xml")
+    try:
+        result = subprocess.run(['hwp5xml', input_path], capture_output=True, timeout=120)
+        if result.returncode == 0 and result.stdout:
+            with open(output_path, 'wb') as f:
+                f.write(result.stdout)
+            return output_path, None
+    except:
+        pass
+    return None, "XML 변환 실패"
+
 # ============== LLM API (Groq 라이브러리 사용) ==============
 def call_groq_api_stream(messages: List[Dict]) -> Generator[str, None, None]:
-    """Groq API 스트리밍 호출 - openai/gpt-oss-120b 모델 사용"""
+    """Groq API 스트리밍 호출"""
     if not GROQ_AVAILABLE:
         yield "❌ Groq 라이브러리가 설치되지 않았습니다. pip install groq"
         return
@@ -1107,12 +1129,9 @@ def chat_response(message: str, history: List[Dict], file: Optional[str],
     
     # 디버그 로그
     print(f"\n🤖 [API 요청]")
-    print(f"  - 모델: openai/gpt-oss-120b")
     print(f"  - 메시지 수: {len(api_messages)}")
     print(f"  - 파일 타입: {file_type}")
     print(f"  - 문서 길이: {len(file_content) if file_content else 0} 글자")
-    if file_content:
-        print(f"  - 문서 미리보기: {file_content[:200]}...")
     
     # 응답 생성
     full_response = ""
@@ -1143,34 +1162,26 @@ def load_session(session_id: str) -> tuple:
     messages = get_session_messages(session_id, limit=50)
     return [{"role": m["role"], "content": m["content"]} for m in messages], session_id
 
-# ============== HWP 변환기 ==============
-def convert_to_odt_subprocess(input_path, output_dir):
-    output_path = os.path.join(output_dir, "output.odt")
-    try:
-        result = subprocess.run(['hwp5odt', '--output', output_path, input_path], capture_output=True, timeout=120)
-        if result.returncode == 0 and os.path.exists(output_path):
-            return output_path, None
-    except:
-        pass
-    return None, "ODT 변환 실패"
-
-def convert_to_xml_subprocess(input_path, output_dir):
-    output_path = os.path.join(output_dir, "output.xml")
-    try:
-        result = subprocess.run(['hwp5xml', input_path], capture_output=True, timeout=120)
-        if result.returncode == 0 and result.stdout:
-            with open(output_path, 'wb') as f:
-                f.write(result.stdout)
-            return output_path, None
-    except:
-        pass
-    return None, "XML 변환 실패"
-
+# ============== HWP 변환기 (수정됨) ==============
 def convert_hwp(file, output_format, progress=gr.Progress()):
-    if not file:
+    """HWP/HWPX 파일을 다양한 형식으로 변환 (다운로드 문제 수정)"""
+    if file is None:
         return None, "❌ 파일을 업로드해주세요.", ""
     
-    input_file = file.name if hasattr(file, 'name') else str(file)
+    # Gradio 버전에 따른 파일 경로 처리
+    if isinstance(file, str):
+        input_file = file
+    elif hasattr(file, 'name'):
+        input_file = file.name
+    else:
+        input_file = str(file)
+    
+    print(f"\n🔄 [변환 시작] 입력 파일: {input_file}")
+    
+    # 파일 존재 확인
+    if not os.path.exists(input_file):
+        return None, f"❌ 파일을 찾을 수 없습니다: {input_file}", ""
+    
     ext_lower = Path(input_file).suffix.lower()
     
     if ext_lower not in ['.hwp', '.hwpx']:
@@ -1178,19 +1189,25 @@ def convert_hwp(file, output_format, progress=gr.Progress()):
     
     progress(0.1, desc="📖 파일 읽는 중...")
     version, is_valid = check_hwp_version(input_file)
+    print(f"  파일 버전: {version}, 유효: {is_valid}")
+    
     if not is_valid:
         return None, f"❌ 지원하지 않는 파일: {version}", ""
     
-    tmp_dir = tempfile.mkdtemp()
+    # 임시 디렉토리 생성
+    tmp_dir = tempfile.mkdtemp(prefix="hwp_convert_")
+    print(f"  임시 디렉토리: {tmp_dir}")
     
     try:
         input_filename = os.path.basename(input_file)
         input_path = os.path.join(tmp_dir, input_filename)
-        shutil.copy(input_file, input_path)
+        shutil.copy2(input_file, input_path)
         
         progress(0.3, desc=f"🔄 {output_format}로 변환 중...")
         
-        output_path, error, ext = None, None, ""
+        output_path = None
+        error = None
+        ext = ""
         
         if output_format == "HTML":
             if ext_lower == '.hwpx':
@@ -1209,19 +1226,21 @@ def convert_hwp(file, output_format, progress=gr.Progress()):
             
         elif output_format == "TXT (텍스트)":
             text, error = extract_text_from_hwp_or_hwpx(input_path)
-            if text:
+            if text and text.strip():
                 output_path = os.path.join(tmp_dir, "output.txt")
                 with open(output_path, 'w', encoding='utf-8') as f:
                     f.write(text)
-            ext = ".txt"
+                ext = ".txt"
+                print(f"  TXT 생성 완료: {len(text)} 글자")
             
         elif output_format == "⭐ MARKDOWN (추천)":
             text, error = convert_hwp_to_markdown(input_path)
-            if text:
+            if text and text.strip():
                 output_path = os.path.join(tmp_dir, "output.md")
                 with open(output_path, 'w', encoding='utf-8') as f:
                     f.write(text)
-            ext = ".md"
+                ext = ".md"
+                print(f"  MD 생성 완료: {len(text)} 글자")
             
         elif output_format == "XML":
             if ext_lower == '.hwpx':
@@ -1232,31 +1251,48 @@ def convert_hwp(file, output_format, progress=gr.Progress()):
                             if name.endswith('.xml'):
                                 with zf.open(name) as f:
                                     xml_contents.append(f"<!-- {name} -->\n{f.read().decode('utf-8', errors='ignore')}")
-                        output_path = os.path.join(tmp_dir, "output.xml")
-                        with open(output_path, 'w', encoding='utf-8') as f:
-                            f.write('\n\n'.join(xml_contents))
+                        if xml_contents:
+                            output_path = os.path.join(tmp_dir, "output.xml")
+                            with open(output_path, 'w', encoding='utf-8') as f:
+                                f.write('\n\n'.join(xml_contents))
+                            ext = ".xml"
                 except Exception as e:
                     error = f"HWPX XML 추출 실패: {e}"
             else:
                 output_path, error = convert_to_xml_subprocess(input_path, tmp_dir)
-            ext = ".xml"
+                ext = ".xml"
         
+        # 변환 결과 확인
         if not output_path:
+            print(f"  ❌ 변환 실패: {error}")
             return None, f"❌ {error or '변환 실패'}", ""
         
         if not os.path.exists(output_path):
+            print(f"  ❌ 출력 파일 없음: {output_path}")
             return None, "❌ 변환된 파일을 찾을 수 없습니다.", ""
         
         progress(0.8, desc="✅ 완료 중...")
         
+        # 최종 파일명 생성
         base_name = Path(input_filename).stem
-        final_output = os.path.join(tmp_dir, f"{base_name}{ext}")
+        final_filename = f"{base_name}{ext}"
+        final_output = os.path.join(tmp_dir, final_filename)
+        
+        # 파일명이 다르면 복사
         if output_path != final_output:
             shutil.copy2(output_path, final_output)
         
+        # 파일 검증
+        if not os.path.exists(final_output):
+            return None, "❌ 최종 파일 생성 실패", ""
+        
         file_size = os.path.getsize(final_output)
+        if file_size == 0:
+            return None, "❌ 변환된 파일이 비어있습니다.", ""
+        
         size_str = f"{file_size/1024:.1f} KB" if file_size > 1024 else f"{file_size} bytes"
         
+        # 미리보기 생성
         preview = ""
         if ext in ['.txt', '.md', '.xml']:
             try:
@@ -1264,24 +1300,35 @@ def convert_hwp(file, output_format, progress=gr.Progress()):
                     preview = f.read(5000)
                     if len(preview) >= 5000:
                         preview += "\n\n... (생략)"
-            except:
-                pass
+            except Exception as e:
+                preview = f"미리보기 로드 실패: {e}"
         elif ext == '.zip':
             preview = "📦 HTML이 ZIP으로 압축되었습니다."
+        elif ext == '.html':
+            preview = "🌐 HTML 파일이 생성되었습니다."
+        elif ext == '.odt':
+            preview = "📄 ODT 파일이 생성되었습니다."
         
         progress(1.0, desc="🎉 완료!")
-        return final_output, f"✅ 변환 완료: {base_name}{ext} ({size_str})", preview
+        
+        print(f"  ✅ 변환 완료: {final_output}")
+        print(f"     크기: {size_str}")
+        print(f"     존재: {os.path.exists(final_output)}")
+        
+        # 파일 경로 반환
+        return final_output, f"✅ 변환 완료: {final_filename} ({size_str})", preview
         
     except Exception as e:
         import traceback
         traceback.print_exc()
         return None, f"❌ 오류: {str(e)}", ""
 
+
 # ============== Gradio UI ==============
 with gr.Blocks(
     title="HWPower AI 어시스턴트", 
     css=COMIC_CSS, 
-    delete_cache=(60, 60),
+    delete_cache=(3600, 3600),  # 1시간 캐시 유지
     head="""
 <style>
 header, .container.svelte-1kyws56, #huggingface-space-header, 
@@ -1346,8 +1393,6 @@ setInterval(hideHfHeader, 300);
     with gr.Tabs():
         # Tab 1: AI 채팅
         with gr.Tab("💬 AI 채팅"):
-            # Feature Box
-            
             with gr.Row():
                 with gr.Column(scale=1):
                     gr.HTML("""
@@ -1366,7 +1411,7 @@ setInterval(hideHfHeader, 300);
                         refresh_btn = gr.Button("🔄 새로고침", size="sm")
                 
                 with gr.Column(scale=3):
-                    chatbot = gr.Chatbot(label="💬 AI 대화", height=500)
+                    chatbot = gr.Chatbot(label="💬 AI 대화", height=500, type="messages")
                     
                     with gr.Row():
                         file_upload = gr.File(
@@ -1483,19 +1528,31 @@ setInterval(hideHfHeader, 300);
         for r in chat_response(msg, hist, f, sid):
             yield r[0], r[1], "", None
     
-    submit_btn.click(on_submit, [msg_input, chatbot, file_upload, session_state],
-                     [chatbot, session_state, msg_input, file_upload])
-    msg_input.submit(on_submit, [msg_input, chatbot, file_upload, session_state],
-                     [chatbot, session_state, msg_input, file_upload])
+    submit_btn.click(
+        fn=on_submit, 
+        inputs=[msg_input, chatbot, file_upload, session_state],
+        outputs=[chatbot, session_state, msg_input, file_upload]
+    )
+    msg_input.submit(
+        fn=on_submit, 
+        inputs=[msg_input, chatbot, file_upload, session_state],
+        outputs=[chatbot, session_state, msg_input, file_upload]
+    )
     
-    new_btn.click(lambda: ([], create_session(), None, ""), outputs=[chatbot, session_state, file_upload, msg_input])
-    clear_btn.click(lambda: ([], None, ""), outputs=[chatbot, file_upload, msg_input])
+    new_btn.click(
+        fn=lambda: ([], create_session(), None, ""), 
+        outputs=[chatbot, session_state, file_upload, msg_input]
+    )
+    clear_btn.click(
+        fn=lambda: ([], None, ""), 
+        outputs=[chatbot, file_upload, msg_input]
+    )
     
     def refresh():
         sessions = get_all_sessions()
         return [[s["session_id"][:8], s["title"] or "제목없음", s["updated_at"][:16] if s["updated_at"] else ""] for s in sessions]
     
-    refresh_btn.click(refresh, outputs=[session_list])
+    refresh_btn.click(fn=refresh, outputs=[session_list])
     
     def select_session(evt: gr.SelectData, data):
         if evt.index[0] < len(data):
@@ -1504,9 +1561,16 @@ setInterval(hideHfHeader, 300);
                     return load_session(s["session_id"])
         return [], ""
     
-    session_list.select(select_session, [session_list], [chatbot, session_state])
-    convert_btn.click(convert_hwp, [hwp_input, format_select], [file_out, status_out, preview_out])
-    demo.load(refresh, outputs=[session_list])
+    session_list.select(fn=select_session, inputs=[session_list], outputs=[chatbot, session_state])
+    
+    # 변환 버튼 이벤트 (수정됨)
+    convert_btn.click(
+        fn=convert_hwp, 
+        inputs=[hwp_input, format_select], 
+        outputs=[file_out, status_out, preview_out]
+    )
+    
+    demo.load(fn=refresh, outputs=[session_list])
 
 if __name__ == "__main__":
     demo.launch(ssr_mode=False)