Update app.py

app.py

@@ -37,7 +37,7 @@ def check_flagged_address(address: str) -> str:
 # Define the tool for the agent
 flagged_address_tool = Tool(
     name="check_flagged_address",
-    description="Checks if a cryptocurrency address is flagged in the Bybit hack database.",
+    description="Checks if a cryptocurrency address is flagged in the Bybit hack database and identifies its blockchain.",
     fn=check_flagged_address
 )
 
@@ -48,13 +48,14 @@ def chat_with_agent(user_input, chat_history, hf_token):
     
     # Check if token is provided
     if not hf_token or hf_token.strip() == "":
-        return chat_history + [[user_input, "Please enter a valid Hugging Face API token."]], ""
+        chat_history.append({"role": "user", "content": user_input})
+        chat_history.append({"role": "assistant", "content": "Please enter a valid Hugging Face API token."})
+        return chat_history, ""
     
     # Initialize the CodeAgent with the user-provided HF token
     agent = CodeAgent(
         model=HfApiModel(model_id="mistralai/Mixtral-8x7B-Instruct-v0.1", token=hf_token),
-        tools=[flagged_address_tool],
-        system_prompt="You are an agent that checks if cryptocurrency addresses are flagged as part of the Bybit hack. Given an address, use the check_flagged_address tool to verify its status and report the result, including the blockchain it's on."
+        tools=[flagged_address_tool]
     )
     
     # Run the agent with the user's input
@@ -63,8 +64,9 @@ def chat_with_agent(user_input, chat_history, hf_token):
     except Exception as e:
         response = f"Error: {str(e)}. Check your HF token or try again."
     
-    # Append as a [user, bot] pair for Chatbot
-    chat_history.append([user_input, response])
+    # Append as OpenAI-style messages
+    chat_history.append({"role": "user", "content": user_input})
+    chat_history.append({"role": "assistant", "content": response})
     
     # Return updated history and clear the input box
     return chat_history, ""
@@ -87,8 +89,8 @@ with gr.Blocks(title="Bybit Hack Address Checker") as demo:
                 lines=1
             )
             
-            # Chatbot component
-            chatbot = gr.Chatbot(label="Conversation")
+            # Chatbot component with messages format
+            chatbot = gr.Chatbot(label="Conversation", type="messages")
             
             # Address input
             msg = gr.Textbox(