Update app.py

app.py

@@ -1224,10 +1224,10 @@ Provide patched code and explain what changed.
 
 async def centaur_chat_agent(user_msg, chat_history, target_id, state,
                               current_ep, current_pl, current_code, http_method, api_key):
+    # Gradio 6: Chatbot history is [[user, bot], ...] tuples
+    chat_history = chat_history or []
     if not api_key or not target_id:
-        chat_history = chat_history or []
-        chat_history.append({"role": "user", "content": user_msg})
-        chat_history.append({"role": "assistant", "content": "🔴 Missing API Key or Target Selection."})
+        chat_history.append([user_msg, "🔴 Missing API Key or Target Selection."])
         return chat_history, current_ep, current_pl, current_code, http_method, state
 
     llm = genai.Client(api_key=api_key)
@@ -1285,9 +1285,7 @@ INSTRUCTIONS:
         data = CentaurChatResponse.model_validate_json(resp.text)
 
         formatted_reply = f"**{data.chat_reply}**\n\n📋 Technical Analysis:\n{data.technical_analysis}"
-        chat_history = chat_history or []
-        chat_history.append({"role": "user", "content": user_msg})
-        chat_history.append({"role": "assistant", "content": formatted_reply})
+        chat_history.append([user_msg, formatted_reply])
 
         state["arsenal"][target_id].append({
             "Timestamp": datetime.now().strftime("%H:%M:%S"),
@@ -1302,9 +1300,7 @@ INSTRUCTIONS:
                 data.suggested_code, data.http_method, state)
     except Exception as e:
         exploit_logger.log("ERROR", f"Chat agent failed: {e}")
-        chat_history = chat_history or []
-        chat_history.append({"role": "user", "content": user_msg})
-        chat_history.append({"role": "assistant", "content": f"⚠️ Agent Error: {str(e)}"})
+        chat_history.append([user_msg, f"⚠️ Agent Error: {str(e)}"])
         return chat_history, current_ep, current_pl, current_code, http_method, state
 
 # ==============================================================================
@@ -1627,11 +1623,13 @@ async def patch_code_with_ai(current_code, user_request, tid, state, api_key):
 # GRADIO UI  — all gr.Markdown() scale= calls removed (not supported)
 # ==============================================================================
 
-with gr.Blocks(theme=gr.themes.Monochrome(), css="""
+_CSS = """
     .primary-btn { background: linear-gradient(45deg,#ff0080,#ff8c00) !important; }
     .exploit-table { font-family: 'Courier New', monospace; font-size: 0.9em; }
     .code-editor   { font-family: 'Courier New', monospace; font-size: 0.95em; }
-""") as demo:
+"""
+
+with gr.Blocks() as demo:
 
     engine_state = gr.State({"profiles": {}, "arsenal": defaultdict(list)})
 
@@ -1734,7 +1732,7 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css="""
                     gr.Markdown("### 🤖 AI Agent — CENTAUR")
                     chatbot = gr.Chatbot(
                         height=400, label="Conversational Interface",
-                        show_label=False, type="messages",
+                        show_label=False,
                         avatar_images=(None, "https://em-content.zobj.net/source/twitter/376/robot_1f916.png")
                     )
                     chat_in = gr.Textbox(
@@ -1840,4 +1838,5 @@ Deploy AI agents to: analyse vulnerabilities → generate strategies → craft p
     )
 
 
-demo.launch(share=False, server_name="0.0.0.0", server_port=7860)
+demo.launch(share=False, server_name="0.0.0.0", server_port=7860,
+            theme=gr.themes.Monochrome(), css=_CSS)