Update ui_layout.py

ui_layout.py

@@ -3,32 +3,38 @@ import gradio as gr
 def create_ui(process_fn):
     """
     The AI can rewrite this function to change the UI.
-    It MUST return a gr.Blocks object.
-    It MUST attach the 'process_fn' to the chat interface.
+    Updated for Gradio 5+ (Messages format).
     """
-    with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    with gr.Blocks() as demo:
         gr.Markdown("# 🛡️ Self-Modifying Isolated AI")
         gr.Markdown("I run on CPU. I can install tools and change this UI.")
         
-        with gr.Row():
-            chatbot = gr.Chatbot(height=400)
+        # type="messages" is REQUIRED for Gradio 5
+        chatbot = gr.Chatbot(height=400, type="messages")
         
         msg = gr.Textbox(label="Command or Chat")
         clear = gr.Button("Clear")
 
-        # Standard Chat Interface Logic
         def user(user_message, history):
-            return "", history + [[user_message, None]]
+            # Append user message in the new dictionary format
+            return "", history + [{"role": "user", "content": user_message}]
 
         def bot(history):
-            user_message = history[-1][0]
+            # Get the content of the last message (User's message)
+            user_message = history[-1]["content"]
+            
+            # Call the AI process function
             bot_message = process_fn(user_message)
-            history[-1][1] = bot_message
+            
+            # Append the assistant's response
+            history.append({"role": "assistant", "content": bot_message})
             return history
 
         msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
             bot, chatbot, chatbot
         )
-        clear.click(lambda: None, None, chatbot, queue=False)
+        
+        # Clear button resets history to an empty list
+        clear.click(lambda: [], None, chatbot, queue=False)
         
     return demo