Update app.py

app.py

@@ -103,14 +103,24 @@ with gr.Blocks() as demo:
     ascii_toggle = gr.Checkbox(label="Convert image to ASCII art")
 
     def respond(msg, chat_history, mode, model, img, ascii_on):
-        # 1. Get the bot's response
+        # 1. Get the bot's response (this includes the ASCII art in the message string)
         bot_message = chat_fn(msg, chat_history, mode, model, img, ascii_on)
         
-        # 2. Append the new interaction as a list of messages
+        # 2. Append the new interaction to the chat history
+        # First, add the ASCII art as a separate, formatted block if it exists
+        if img and ascii_on:
+            ascii_art = image_to_ascii(img)
+            if ascii_art:
+                # Use gr.Code to display the ASCII art with proper formatting
+                chat_history.append({"role": "user", "content": gr.Code(ascii_art, language="text")})
+        
+        # Then, add the user's text message
         chat_history.append({"role": "user", "content": msg})
+        
+        # Finally, add the assistant's response
         chat_history.append({"role": "assistant", "content": bot_message})
         
-        # 3. Return history and clear the input box
+        # 3. Return the updated history and clear the input box
         return chat_history, ""
 
     send_btn.click(