Update app.py

app.py

@@ -251,9 +251,9 @@ def process_new_user_message(message: dict) -> list[dict]:
         return content_list
 
     # 7) 이미지 처리
-    if "<image>" in message["text"]:
+    if "<image>" in message["text"] and image_files:  # 이미지 파일이 있는 경우에만
         # interleaved
-        return process_interleaved_images(message)
+        return process_interleaved_images({"text": message["text"], "files": image_files})
     else:
         # 일반 여러 장
         for img_path in image_files:
@@ -261,7 +261,6 @@ def process_new_user_message(message: dict) -> list[dict]:
 
     return content_list
 
-
 ##################################################
 # history -> LLM 메시지 변환
 ##################################################
@@ -300,8 +299,26 @@ def run(message: dict, history: list[dict], system_prompt: str = "", max_new_tok
     if system_prompt:
         messages.append({"role": "system", "content": [{"type": "text", "text": system_prompt}]})
     messages.extend(process_history(history))
-    messages.append({"role": "user", "content": process_new_user_message(message)})
+    
+    # 사용자 메시지 처리
+    user_content = process_new_user_message(message)
+    
+    # 이미지가 아닌 파일들만 텍스트로 변환
+    processed_content = []
+    for item in user_content:
+        if item["type"] == "image":
+            # 이미지 파일인지 확인
+            if re.search(r"\.(png|jpg|jpeg|gif|webp)$", item["url"], re.IGNORECASE):
+                processed_content.append(item)
+            else:
+                # 이미지가 아닌 파일은 텍스트로 변환
+                processed_content.append({"type": "text", "text": f"[File: {os.path.basename(item['url'])}]"})
+        else:
+            processed_content.append(item)
+    
+    messages.append({"role": "user", "content": processed_content})
 
+    # LLM 처리는 그대로 진행
     inputs = processor.apply_chat_template(
         messages,
         add_generation_prompt=True,
@@ -324,7 +341,6 @@ def run(message: dict, history: list[dict], system_prompt: str = "", max_new_tok
         output += new_text
         yield output
 
-
 ##################################################
 # 예시들 (기존)
 ##################################################