Update app.py

app.py

@@ -141,6 +141,25 @@ def handle_file_upload(file_path, session_id):
     return file_path, json.dumps(status, indent=2), session_id
 
 
+def format_chat_history(history, new_user_msg, new_assistant_msg):
+    """
+    Convert chat history to new Gradio format (list of dicts with role/content)
+    Gradio 5.x expects: [{"role": "user", "content": "..."}, {"role": "assistant", "content": "..."}]
+    """
+    messages = []
+    
+    # Add existing history
+    for user_msg, bot_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_msg})
+    
+    # Add new messages
+    messages.append({"role": "user", "content": new_user_msg})
+    messages.append({"role": "assistant", "content": new_assistant_msg})
+    
+    return messages
+
+
 def chatbot_response_streaming(message: str, history: List, session_id: str, file_path: str = None):
     """
     Handle chat messages with streaming updates
@@ -175,12 +194,12 @@ def chatbot_response_streaming(message: str, history: List, session_id: str, fil
             }
             response_text = f"```json\n{json.dumps(response, indent=2)}\n```"
             session_manager.add_message(session_id, "assistant", response_text)
-            yield history + [[message, response_text]]
+            yield format_chat_history(history, message, response_text)
             return
         
         try:
             # Generate pipeline using Bedrock → Gemini fallback
-            yield history + [[message, "🤖 Generating pipeline with AI...\n⏳ Trying Bedrock first..."]]
+            yield format_chat_history(history, message, "🤖 Generating pipeline with AI...\n⏳ Trying Bedrock first...")
             
             pipeline = generate_pipeline(
                 user_input=message,
@@ -201,7 +220,7 @@ def chatbot_response_streaming(message: str, history: List, session_id: str, fil
             response_text = formatted_display + f"\n\n```json\n{json.dumps(pipeline, indent=2)}\n```"
             
             session_manager.add_message(session_id, "assistant", response_text)
-            yield history + [[message, response_text]]
+            yield format_chat_history(history, message, response_text)
             return
             
         except Exception as e:
@@ -213,7 +232,7 @@ def chatbot_response_streaming(message: str, history: List, session_id: str, fil
             }
             response_text = f"```json\n{json.dumps(error_response, indent=2)}\n```"
             session_manager.add_message(session_id, "assistant", response_text)
-            yield history + [[message, response_text]]
+            yield format_chat_history(history, message, response_text)
             return
     
     # ========================
@@ -237,7 +256,7 @@ def chatbot_response_streaming(message: str, history: List, session_id: str, fil
                 "steps": []
             }
             accumulated_response = f"```json\n{json.dumps(initial_status, indent=2)}\n```"
-            yield history + [[message, accumulated_response]]
+            yield format_chat_history(history, message, accumulated_response)
             
             steps_completed = []
             final_payload = None
@@ -261,7 +280,7 @@ def chatbot_response_streaming(message: str, history: List, session_id: str, fil
                             "executor": event.get("executor", "unknown")
                         }
                         accumulated_response = f"```json\n{json.dumps(info_status, indent=2)}\n```"
-                        yield history + [[message, accumulated_response]]
+                        yield format_chat_history(history, message, accumulated_response)
                     
                     # Step updates
                     elif event_type == "step":
@@ -282,7 +301,7 @@ def chatbot_response_streaming(message: str, history: List, session_id: str, fil
                             "steps_completed": steps_completed
                         }
                         accumulated_response = f"```json\n{json.dumps(progress_status, indent=2)}\n```"
-                        yield history + [[message, accumulated_response]]
+                        yield format_chat_history(history, message, accumulated_response)
                     
                     # Final result
                     elif event_type == "final":
@@ -300,7 +319,7 @@ def chatbot_response_streaming(message: str, history: List, session_id: str, fil
                         final_response = f"```json\n{json.dumps(error_result, indent=2)}\n```"
                         session_manager.update_session(session_id, {"state": ConversationState.INITIAL})
                         session_manager.add_message(session_id, "assistant", final_response)
-                        yield history + [[message, final_response]]
+                        yield format_chat_history(history, message, final_response)
                         return
                 
                 # Process final result
@@ -336,7 +355,7 @@ def chatbot_response_streaming(message: str, history: List, session_id: str, fil
                     session_manager.update_session(session_id, {"state": ConversationState.INITIAL})
                 
                 session_manager.add_message(session_id, "assistant", final_response)
-                yield history + [[message, final_response]]
+                yield format_chat_history(history, message, final_response)
                 return
                 
             except Exception as e:
@@ -349,7 +368,7 @@ def chatbot_response_streaming(message: str, history: List, session_id: str, fil
                 final_response = f"```json\n{json.dumps(error_result, indent=2)}\n```"
                 session_manager.update_session(session_id, {"state": ConversationState.INITIAL})
                 session_manager.add_message(session_id, "assistant", final_response)
-                yield history + [[message, final_response]]
+                yield format_chat_history(history, message, final_response)
                 return
         
         # REJECT - Cancel the pipeline
@@ -365,7 +384,7 @@ def chatbot_response_streaming(message: str, history: List, session_id: str, fil
             }
             response = f"```json\n{json.dumps(response_data, indent=2)}\n```"
             session_manager.add_message(session_id, "assistant", response)
-            yield history + [[message, response]]
+            yield format_chat_history(history, message, response)
             return
         
         # EDIT - Request modifications
@@ -385,7 +404,7 @@ def chatbot_response_streaming(message: str, history: List, session_id: str, fil
             }
             response = f"```json\n{json.dumps(edit_help, indent=2)}\n```"
             session_manager.add_message(session_id, "assistant", response)
-            yield history + [[message, response]]
+            yield format_chat_history(history, message, response)
             return
         
         # Try to modify pipeline based on user input
@@ -410,7 +429,7 @@ def chatbot_response_streaming(message: str, history: List, session_id: str, fil
                     formatted = format_pipeline_for_display(new_pipeline)
                     response = formatted + f"\n\n```json\n{json.dumps(new_pipeline, indent=2)}\n```"
                     session_manager.add_message(session_id, "assistant", response)
-                    yield history + [[message, response]]
+                    yield format_chat_history(history, message, response)
                     return
                     
                 except Exception as e:
@@ -422,7 +441,7 @@ def chatbot_response_streaming(message: str, history: List, session_id: str, fil
                     }
                     response = f"```json\n{json.dumps(error_response, indent=2)}\n```"
                     session_manager.add_message(session_id, "assistant", response)
-                    yield history + [[message, response]]
+                    yield format_chat_history(history, message, response)
                     return
             
             # Default waiting message
@@ -433,13 +452,13 @@ def chatbot_response_streaming(message: str, history: List, session_id: str, fil
             }
             response = f"```json\n{json.dumps(response_data, indent=2)}\n```"
             session_manager.add_message(session_id, "assistant", response)
-            yield history + [[message, response]]
+            yield format_chat_history(history, message, response)
             return
     
     # Default fallback
     response = json.dumps({"status": "ready", "message": "Ready for your next instruction"}, indent=2)
     session_manager.add_message(session_id, "assistant", response)
-    yield history + [[message, response]]
+    yield format_chat_history(history, message, response)
 
 
 # ========================
@@ -462,8 +481,8 @@ with gr.Blocks(title="MasterLLM v2.0 - AI Pipeline Orchestrator") as demo:
     
     with gr.Row():
         with gr.Column(scale=3):
-            # Chat interface (simplified for older Gradio)
-            chatbot = gr.Chatbot(label="Chat")
+            # Chat interface (Gradio 5.x with messages format)
+            chatbot = gr.Chatbot(label="Chat", type="messages")
             
             # Text input
             msg = gr.Textbox(