Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import gradio as gr
 from gradio import ChatMessage
-from typing import Iterator, List, Dict, Tuple
+from typing import Iterator, List, Dict, Tuple, Any
 import google.generativeai as genai
 from huggingface_hub import HfApi
 import requests
@@ -169,24 +169,33 @@ def explain_usage(app_content: str) -> str:
     except Exception as e:
         return f"사용법 설명 생성 중 오류 발생: {str(e)}"
 
-def convert_chat_history(messages: List[ChatMessage]) -> List[Tuple[str, str]]:
+def convert_chat_history(messages: List[Any]) -> List[Tuple[str, str]]:
     """
-    ChatMessage 객체 리스트를 (user, assistant) 튜플 목록으로 변환합니다.
+    메시지 목록의 각 항목이 ChatMessage 객체라면 (user, assistant) 튜플로,
+    이미 튜플인 경우 그대로 반환합니다.
     """
     conv = []
     i = 0
     while i < len(messages):
-        if messages[i].role == "user":
-            user_text = messages[i].content
-            bot_text = ""
-            if i + 1 < len(messages) and messages[i+1].role == "assistant":
-                bot_text = messages[i+1].content
-                i += 2
+        # 만약 이미 튜플이면 그대로 사용
+        if isinstance(messages[i], tuple):
+            conv.append(messages[i])
+            i += 1
+        # 만약 ChatMessage 객체라면
+        elif hasattr(messages[i], "role"):
+            if messages[i].role == "user":
+                user_text = messages[i].content
+                bot_text = ""
+                if i + 1 < len(messages) and hasattr(messages[i+1], "role") and messages[i+1].role == "assistant":
+                    bot_text = messages[i+1].content
+                    i += 2
+                else:
+                    i += 1
+                conv.append((user_text, bot_text))
             else:
+                conv.append(("", messages[i].content))
                 i += 1
-            conv.append((user_text, bot_text))
         else:
-            conv.append(("", messages[i].content))
             i += 1
     return conv