Update app.py

app.py

@@ -138,14 +138,16 @@ def get_messages(client, thread):
     return messages
                         
 def extract_content_values(data):
-    content_values = []
+    text_values, image_value = [], []
     
     for item in data.data:
         for content in item.content:
             if content.type == "text":
-                content_values.append(content.text.value)
+                text_values.append(content.text.value)
+            if content.type == "image_file":
+                image_values.append(content.image_file.file_id)
     
-    return content_values
+    return text_values, image_values
 
 def chat(message, history):
     if not message:
@@ -170,13 +172,15 @@ def chat(message, history):
     
     messages = get_messages(client, thread)
 
-    content_values = extract_content_values(messages)
+    text_values, image_values = extract_content_values(messages)
 
     print("###")
-    print(content_values[0])
+    print(join(text_values))
+    print("###")
+    print(join(image_values))
     print("###")
     
-    return f"{content_values[0]}<br /><br />{download_file}"
+    return {join(text_values)}
 
 gr.ChatInterface(
         fn=chat,