Update app.py

app.py

@@ -175,16 +175,13 @@ def analyze_space(url: str, progress=gr.Progress()):
         print(traceback.format_exc())
         return f"오류가 발생했습니다: {str(e)}", "", None, "", "", "", "", 10
 
-
-
-
 def respond(
-    message,
+    message: str,
     history: List[Tuple[str, str]],
-    system_message="",
-    max_tokens=1024,
-    temperature=0.7,
-    top_p=0.9,
+    system_message: str = "",
+    max_tokens: int = 1024,
+    temperature: float = 0.7,
+    top_p: float = 0.9,
 ):
     system_prefix = """당신은 허깅페이스에 특화된 AI 코딩 전문가입니다. 사용자의 질문에 친절하고 상세하게 답변해주세요. 
     Gradio 특성을 정확히 인식하고 Requirements.txt 누락없이 코딩과 오류를 해결해야 합니다.
@@ -193,7 +190,8 @@ def respond(
     messages = [{"role": "system", "content": f"{system_prefix} {system_message}"}]
     for user_msg, assistant_msg in history:
         messages.append({"role": "user", "content": user_msg})
-        messages.append({"role": "assistant", "content": assistant_msg})
+        if assistant_msg:
+            messages.append({"role": "assistant", "content": assistant_msg})
     messages.append({"role": "user", "content": message})
 
     response = ""
@@ -317,7 +315,8 @@ def create_ui():
                                         )
 
                 with gr.TabItem("AI 코딩"):
-                    chatbot = gr.Chatbot(label="대화", type='messages')
+                    chatbot = gr.Chatbot(label="대화")
+
                     msg = gr.Textbox(label="메시지")
                     
                     # 숨겨진 상태로 파라미터 설정
@@ -340,11 +339,13 @@ def create_ui():
                         bot_message = ""
                         for response in respond(message, chat_history, max_tokens=max_tokens, temperature=temperature, top_p=top_p):
                             bot_message = response  # 마지막 응답을 저장
+                            yield "", chat_history + [(message, bot_message)]
     
-                        chat_history.append({"role": "user", "content": message})
-                        chat_history.append({"role": "assistant", "content": bot_message})
+                        chat_history.append((message, bot_message))
                         return "", chat_history
 
+
+
                     msg.submit(respond_wrapper, [msg, chatbot, max_tokens, temperature, top_p], [msg, chatbot])
 
             space_id_state = gr.State()