Update app.py

app.py

@@ -4,18 +4,16 @@ import requests
 import json
 
 def create_deepseek_interface():
-    # 채팅 기록 초기화 (메시지 형식)
-    chat_history = []
-    
-    # DeepSeek API 호출 함수
+    # 채팅 기록 상태를 저장할 State 객체
     def query_deepseek(message, history, api_key):
         if not api_key:
             return history, "Fireworks AI API 키를 입력해주세요"
         
         # API 요청을 위한 대화 기록 준비
         messages = []
-        for h in history:
-            messages.append(h)
+        for user, assistant in history:
+            messages.append({"role": "user", "content": user})
+            messages.append({"role": "assistant", "content": assistant})
         
         # 새 사용자 메시지 추가
         messages.append({"role": "user", "content": message})
@@ -47,14 +45,12 @@ def create_deepseek_interface():
             result = response.json()
             assistant_response = result.get("choices", [{}])[0].get("message", {}).get("content", "")
             
-            # 대화 기록 업데이트 (메시지 형식으로)
-            new_history = history.copy()
-            new_history.append({"role": "user", "content": message})
-            new_history.append({"role": "assistant", "content": assistant_response})
-            return new_history, ""
+            # 대화 기록 업데이트 (Gradio chatbot 형식으로)
+            history.append((message, assistant_response))
+            return history, ""
         except requests.exceptions.RequestException as e:
             error_msg = f"API 오류: {str(e)}"
-            if hasattr(response, 'status_code') and response.status_code == 401:
+            if hasattr(e, 'response') and e.response and e.response.status_code == 401:
                 error_msg = "인증 실패. API 키를 확인해주세요."
             return history, error_msg
     
@@ -68,9 +64,6 @@ def create_deepseek_interface():
             """
         )
         
-        # 오류 메시지 상태
-        error_state = gr.State("")
-        
         # 메인 레이아웃 (두 개의 열)
         with gr.Row():
             # 사이드바 - 모델 정보 및 API 키
@@ -115,8 +108,7 @@ def create_deepseek_interface():
                 chatbot = gr.Chatbot(
                     height=500,
                     show_label=False,
-                    container=True,
-                    type="messages"
+                    container=True
                 )
                 
                 # 입력 영역
@@ -146,22 +138,20 @@ def create_deepseek_interface():
         # 폼 제출 처리
         def process_query(message, history, api_key):
             if not message.strip():
-                return history
+                return history, ""
             
             updated_history, error = query_deepseek(message, history, api_key)
             
             if error:
-                error_box.value = f"**오류:** {error}"
-                return history
+                return history, f"**오류:** {error}"
             else:
-                error_box.value = ""
-                return updated_history
+                return updated_history, ""
         
         # 버튼과 기능 연결
         submit.click(
             process_query,
             inputs=[msg, chatbot, api_key],
-            outputs=[chatbot]
+            outputs=[chatbot, error_box]
         ).then(
             lambda: "",
             None,
@@ -172,7 +162,7 @@ def create_deepseek_interface():
         msg.submit(
             process_query,
             inputs=[msg, chatbot, api_key],
-            outputs=[chatbot]
+            outputs=[chatbot, error_box]
         ).then(
             lambda: "",
             None,