Update app.py

app.py

@@ -785,12 +785,13 @@ def process_file(file_path: str) -> tuple:
 
 def chat_response(
     message: str,
-    history: List[List[str]],
+    history: List[Dict],
     file: Optional[str],
     session_id: str,
     groq_api_key: str,
     fireworks_api_key: str
 ) -> Generator[tuple, None, None]:
+    """채팅 응답 생성 (Gradio 6.0 messages 형식)"""
     
     if not message.strip() and file is None:
         yield history, session_id
@@ -812,24 +813,37 @@ def chat_response(
         })
         
         if file_type == "error":
-            history = history + [[message, f"❌ {file_content}"]]
+            history = history + [
+                {"role": "user", "content": message or "파일 업로드"},
+                {"role": "assistant", "content": f"❌ {file_content}"}
+            ]
             yield history, session_id
             return
         elif file_type == "unsupported":
-            history = history + [[message, f"⚠️ {file_content}"]]
+            history = history + [
+                {"role": "user", "content": message or "파일 업로드"},
+                {"role": "assistant", "content": f"⚠️ {file_content}"}
+            ]
             yield history, session_id
             return
     
+    # 사용자 메시지 구성
     user_display_message = message
     if file is not None:
         filename = os.path.basename(file)
         user_display_message = f"📎 {filename}\n\n{message}" if message else f"📎 {filename}"
     
-    history = history + [[user_display_message, ""]]
+    # 히스토리에 사용자 메시지와 빈 어시스턴트 메시지 추가
+    history = history + [
+        {"role": "user", "content": user_display_message},
+        {"role": "assistant", "content": ""}
+    ]
     yield history, session_id
     
+    # DB에서 이전 대화 가져오기
     db_messages = get_session_messages(session_id, limit=10)
     
+    # API 메시지 구성
     api_messages = []
     
     api_messages.append({
@@ -857,17 +871,20 @@ def chat_response(
     if file_type == "image":
         for chunk in call_fireworks_api_stream(api_messages, file_content, file_mime, fireworks_api_key):
             full_response += chunk
-            history[-1][1] = full_response
+            # 마지막 메시지(assistant) 업데이트
+            history[-1] = {"role": "assistant", "content": full_response}
             yield history, session_id
     else:
         for chunk in call_groq_api_stream(api_messages, groq_api_key):
             full_response += chunk
-            history[-1][1] = full_response
+            history[-1] = {"role": "assistant", "content": full_response}
             yield history, session_id
     
+    # DB에 저장
     save_message(session_id, "user", current_content, file_info)
     save_message(session_id, "assistant", full_response)
     
+    # 첫 메시지면 세션 제목 업데이트
     if len(db_messages) == 0 and message:
         title = message[:50] + "..." if len(message) > 50 else message
         update_session_title(session_id, title)
@@ -881,18 +898,10 @@ def load_session(session_id: str) -> tuple:
         return [], ""
     
     messages = get_session_messages(session_id, limit=50)
+    # Gradio 6.0 messages 형식으로 변환
     history = []
-    
-    i = 0
-    while i < len(messages):
-        if messages[i]["role"] == "user":
-            user_msg = messages[i]["content"]
-            assistant_msg = ""
-            if i + 1 < len(messages) and messages[i + 1]["role"] == "assistant":
-                assistant_msg = messages[i + 1]["content"]
-                i += 1
-            history.append([user_msg, assistant_msg])
-        i += 1
+    for msg in messages:
+        history.append({"role": msg["role"], "content": msg["content"]})
     
     return history, session_id
 
@@ -1141,10 +1150,6 @@ def convert_hwp(file, output_format, progress=gr.Progress()):
 
 # ============== CSS 스타일 ==============
 css = """
-#col-container {
-    margin: 0 auto;
-    max-width: 1200px;
-}
 .upload-box {
     border: 2px dashed #6366f1 !important;
     border-radius: 12px !important;
@@ -1152,7 +1157,6 @@ css = """
 .download-box {
     border: 2px solid #22c55e !important;
     border-radius: 12px !important;
-    background: linear-gradient(135deg, #f0fdf4 0%, #dcfce7 100%) !important;
 }
 .preview-box {
     max-height: 400px;
@@ -1217,6 +1221,7 @@ with gr.Blocks(title="AI 문서 어시스턴트") as demo:
                         refresh_sessions_btn = gr.Button("🔄 새로고침", size="sm")
                 
                 with gr.Column(scale=3):
+                    # Gradio 6.0: Chatbot은 기본적으로 messages 형식 사용
                     chatbot = gr.Chatbot(
                         label="대화",
                         height=500
@@ -1245,71 +1250,73 @@ with gr.Blocks(title="AI 문서 어시스턴트") as demo:
         
         # Tab 2: HWP 변환기
         with gr.Tab("📄 HWP 변환기", id="converter"):
-            with gr.Column(elem_id="col-container"):
-                gr.Markdown("""
-                ### HWP 파일 변환기
-                한글(HWP) 문서를 다양한 형식으로 변환합니다.
-                """)
-                
-                with gr.Row():
-                    with gr.Column(scale=1):
-                        gr.Markdown("#### 📤 파일 업로드")
-                        hwp_file_input = gr.File(
-                            label="HWP 파일 선택",
-                            file_types=[".hwp"],
-                            type="filepath",
-                            elem_classes=["upload-box"]
-                        )
-                        
-                        format_select = gr.Radio(
-                            label="변환 형식",
-                            choices=["HTML", "ODT (OpenDocument)", "TXT (텍스트)", "Markdown", "XML"],
-                            value="TXT (텍스트)",
-                            info="원하는 출력 형식을 선택하세요"
-                        )
-                        
-                        convert_btn = gr.Button("🔄 변환하기", variant="primary", size="lg")
+            gr.Markdown("""
+            ### HWP 파일 변환기
+            한글(HWP) 문서를 다양한 형식으로 변환합니다.
+            """)
+            
+            with gr.Row():
+                with gr.Column(scale=1):
+                    gr.Markdown("#### 📤 파일 업로드")
+                    hwp_file_input = gr.File(
+                        label="HWP 파일 선택",
+                        file_types=[".hwp"],
+                        type="filepath",
+                        elem_classes=["upload-box"]
+                    )
                     
-                    with gr.Column(scale=1):
-                        gr.Markdown("#### 📥 변환 결과")
-                        status_output = gr.Textbox(
-                            label="상태",
-                            interactive=False,
-                            lines=2
-                        )
-                        
-                        file_output = gr.File(
-                            label="다운로드",
-                            elem_classes=["download-box"]
-                        )
+                    format_select = gr.Radio(
+                        label="변환 형식",
+                        choices=["HTML", "ODT (OpenDocument)", "TXT (텍스트)", "Markdown", "XML"],
+                        value="TXT (텍스트)",
+                        info="원하는 출력 형식을 선택하세요"
+                    )
+                    
+                    convert_btn = gr.Button("🔄 변환하기", variant="primary", size="lg")
                 
-                with gr.Accordion("📋 내용 미리보기", open=False):
-                    preview_output = gr.Textbox(
-                        label="",
-                        lines=15,
-                        max_lines=25,
+                with gr.Column(scale=1):
+                    gr.Markdown("#### 📥 변환 결과")
+                    status_output = gr.Textbox(
+                        label="상태",
                         interactive=False,
-                        elem_classes=["preview-box"]
+                        lines=2
                     )
-                
-                gr.Markdown("""
-                ---
-                #### ℹ️ 안내사항
-                
-                | 형식 | 설명 | 용도 |
-                |------|------|------|
-                | **HTML** | 웹 페이지 형식 | 브라우저에서 보기 |
-                | **ODT** | OpenDocument | LibreOffice, Google Docs |
-                | **TXT** | 순수 텍스트 | 텍스트 추출 |
-                | **Markdown** | 마크다운 형식 | GitHub, 노션 |
-                | **XML** | 구조화 데이터 | 데이터 처리 |
-                
-                > ⚠️ HWP v5 형식(한글 2007+)만 지원 | 암호화 파일 불가
-                """)
+                    
+                    file_output = gr.File(
+                        label="다운로드",
+                        elem_classes=["download-box"]
+                    )
+            
+            with gr.Accordion("📋 내용 미리보기", open=False):
+                preview_output = gr.Textbox(
+                    label="",
+                    lines=15,
+                    max_lines=25,
+                    interactive=False,
+                    elem_classes=["preview-box"]
+                )
+            
+            gr.Markdown("""
+            ---
+            #### ℹ️ 안내사항
+            
+            | 형식 | 설명 | 용도 |
+            |------|------|------|
+            | **HTML** | 웹 페이지 형식 | 브라우저에서 보기 |
+            | **ODT** | OpenDocument | LibreOffice, Google Docs |
+            | **TXT** | 순수 텍스트 | 텍스트 추출 |
+            | **Markdown** | 마크다운 형식 | GitHub, 노션 |
+            | **XML** | 구조화 데이터 | 데이터 처리 |
+            
+            > ⚠️ HWP v5 형식(한글 2007+)만 지원 | 암호화 파일 불가
+            """)
     
     # ============== 이벤트 핸들러 ==============
     
     def on_submit(message, history, file, session_id, groq_key, fireworks_key):
+        # history가 None이면 빈 리스트로 초기화
+        if history is None:
+            history = []
         for result in chat_response(message, history, file, session_id, groq_key, fireworks_key):
             yield result[0], result[1], "", None