Update app.py

app.py

@@ -28,7 +28,12 @@ def generate_response(message, history, thinking_mode):
         if isinstance(user_msg, str):
             user_content = [{"type": "text", "text": user_msg}]
         else:
-            text, file_paths = user_msg if isinstance(user_msg, tuple) else (user_msg.get('text', ''), [f['path'] for f in user_msg.get('files', [])])
+            if isinstance(user_msg, tuple):
+                text, file_paths = user_msg
+            else:
+                text = user_msg.get('text', '')
+                files = user_msg.get('files', [])
+                file_paths = [f['path'] if isinstance(f, dict) else f for f in files]
             user_content = []
             img_paths = file_paths if isinstance(file_paths, list) else []
             for path in img_paths:
@@ -49,14 +54,14 @@ def generate_response(message, history, thinking_mode):
     # Current user message
     if isinstance(message, str):
         curr_text = message
-        curr_files = []
+        curr_file_paths = []
     else:
         curr_text = message.get('text', '')
         curr_files = message.get('files', [])
+        curr_file_paths = [f['path'] if isinstance(f, dict) else f for f in curr_files]
     curr_user_content = []
     curr_images = []
-    for f in curr_files:
-        path = f['path']
+    for path in curr_file_paths:
         if path.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.bmp')):
             try:
                 img = Image.open(path)
@@ -104,7 +109,7 @@ def generate_response(message, history, thinking_mode):
     )
 
     # Prepare display for current user message
-    user_display = curr_text if not curr_files else (curr_text, [f['path'] for f in curr_files])
+    user_display = curr_text if not curr_file_paths else (curr_text, curr_file_paths)
     new_history = history + [(user_display, output_text)]
 
     return "", new_history