initial commit

app.py

@@ -7,11 +7,13 @@ from mcp_client import MCPClient
 
 logger = logging.getLogger(__name__)
 
+
 async def initialize_client():
     client = MCPClient()
     await client.initialize()
     return client
 
+
 def launch_ui():
     with gr.Blocks(title="MCP Chatbot UI", fill_height=True, fill_width=True) as demo:
         gr.Markdown("## Dnext Product Catalog AI Assistant")
@@ -19,9 +21,7 @@ def launch_ui():
 
         chatbot = gr.Chatbot(height=600, label="Chatbot", type="messages")
         msg = gr.Textbox(
-            label="Enter your request",
-            placeholder="Type your message here...",
-            lines=2
+            label="Enter your request", placeholder="Type your message here...", lines=2
         )
         submit_btn = gr.Button("Submit")
         clear_btn = gr.Button("Clear Chat")
@@ -30,32 +30,26 @@ def launch_ui():
             bot_response = await client.invoke(user_message)
             return [
                 {"role": "user", "content": user_message},
-                {"role": "assistant", "content": bot_response}
+                {"role": "assistant", "content": bot_response},
             ]
 
         def clear_chat():
             return [], []
 
-        submit_btn.click(
-            fn=respond,
-            inputs=[msg],
-            outputs=[chatbot]
-        )
-        
-        clear_btn.click(
-            fn=clear_chat,
-            inputs=[],
-            outputs=[chatbot, msg]
-        )
+        submit_btn.click(fn=respond, inputs=[msg], outputs=[chatbot])
+
+        clear_btn.click(fn=clear_chat, inputs=[], outputs=[chatbot, msg])
 
     return demo
 
+
 async def main():
     global client
     client = await initialize_client()
     demo = launch_ui()
     demo.launch(share=False)
 
+
 if __name__ == "__main__":
     asyncio.run(main())
 

mcp_client.py

@@ -93,19 +93,22 @@ class MCPClient:
                 )
                 logger.info("Invoking agent...")
                 config = {"configurable": {"thread_id": "conversation_123"}}
-                result = await self.agent.ainvoke(input={"messages": input_messages}, config=config)
+                result = await self.agent.ainvoke(
+                    input={"messages": input_messages}, config=config
+                )
                 logger.info(f"Agent result: {result}")
                 logger.info("========================================================")
                 last_message = result["messages"][-1]
                 logger.info(f"Last message: {last_message.content}")
                 return last_message.content
 
-# Agent result: {'messages': 
+
+# Agent result: {'messages':
 # [HumanMessage(content='hi', additional_kwargs={}, response_metadata={}, id='205d9484-c4f0-4e9e-962f-218d2e82bc03'),
-# AIMessage(content='This is just a greeting, so no API call is required. If you have any tasks or requests related to product catalog operations, please let me know how I can assist you!', 
-# additional_kwargs={'refusal': None}, 
-# response_metadata={'token_usage': {'completion_tokens': 37, 'prompt_tokens': 27124, 'total_tokens': 27161, 
-# 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 
-# 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 27008}}, 'model_name': 'gpt-4.1-2025-04-14', 
-# 'system_fingerprint': 'fp_799e4ca3f1', 'id': 'chatcmpl-BejXD72NDlc9UqrJiobIA6tQbuNaj', 'service_tier': 'default', 'finish_reason': 'stop', 'logprobs': None}, id='run--dee9f47d-99f9-40a6-b8c7-241668e6ac38-0', 
-# usage_metadata={'input_tokens': 27124, 'output_tokens': 37, 'total_tokens': 27161, 'input_token_details': {'audio': 0, 'cache_read': 27008}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}            
+# AIMessage(content='This is just a greeting, so no API call is required. If you have any tasks or requests related to product catalog operations, please let me know how I can assist you!',
+# additional_kwargs={'refusal': None},
+# response_metadata={'token_usage': {'completion_tokens': 37, 'prompt_tokens': 27124, 'total_tokens': 27161,
+# 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0},
+# 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 27008}}, 'model_name': 'gpt-4.1-2025-04-14',
+# 'system_fingerprint': 'fp_799e4ca3f1', 'id': 'chatcmpl-BejXD72NDlc9UqrJiobIA6tQbuNaj', 'service_tier': 'default', 'finish_reason': 'stop', 'logprobs': None}, id='run--dee9f47d-99f9-40a6-b8c7-241668e6ac38-0',
+# usage_metadata={'input_tokens': 27124, 'output_tokens': 37, 'total_tokens': 27161, 'input_token_details': {'audio': 0, 'cache_read': 27008}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}