Update app.py

app.py

@@ -178,14 +178,21 @@ def analyze_space(url: str, progress=gr.Progress()):
         return f"오류가 발생했습니다: {str(e)}", "", None, "", "", "", "", 10
 
 
+def format_chat_history(chat_history):
+    formatted = []
+    for message in chat_history:
+        if message["role"] in ["user", "assistant"]:
+            formatted.append(message)
+    return formatted
 
+    
 def respond_stream(message: str, chat_history: List[Dict[str, str]], max_tokens: int, temperature: float, top_p: float):
     system_message = """당신은 허깅페이스에 특화된 AI 코딩 전문가입니다. 사용자의 질문에 친절하고 상세하게 답변해주세요. 
     Gradio 특성을 정확히 인식하고 Requirements.txt 누락없이 코딩과 오류를 해결해야 합니다.
     항상 정확하고 유용한 정보를 제공하도록 노력하세요."""
     
     messages = [{"role": "system", "content": system_message}]
-    messages.extend(chat_history)
+    messages.extend(format_chat_history(chat_history))
     messages.append({"role": "user", "content": message})
     
     try:
@@ -205,7 +212,6 @@ def respond_stream(message: str, chat_history: List[Dict[str, str]], max_tokens:
     except Exception as e:
         yield f"응답 생성 중 오류 발생: {str(e)}"
 
-
 def create_ui():
     try:
         css = """
@@ -314,7 +320,7 @@ def create_ui():
 
 
                 with gr.TabItem("AI 코딩"):
-                    chatbot = gr.Chatbot(label="대화", type='messages')
+                    chatbot = gr.Chatbot(label="대화")
                     msg = gr.Textbox(label="메시지")
             
                     max_tokens = gr.Slider(minimum=1, maximum=8000, value=4000, label="Max Tokens", visible=False)
@@ -333,17 +339,17 @@ def create_ui():
                     gr.Examples(examples, inputs=msg)
 
                     def respond_wrapper(message, chat_history, max_tokens, temperature, top_p):
-                        chat_history.append({"role": "user", "content": message})
+                        chat_history.append((message, ""))
                         bot_message = ""
                         for partial_response in respond_stream(message, chat_history, max_tokens, temperature, top_p):
                             bot_message = partial_response
-                            chat_history[-1] = {"role": "assistant", "content": bot_message}
+                            chat_history[-1] = (message, bot_message)
                             yield "", chat_history
                         return
 
                     msg.submit(respond_wrapper, [msg, chatbot, max_tokens, temperature, top_p], [msg, chatbot])
 
-
+