Let AI run MCP

project/chat.py

@@ -1,4 +1,6 @@
 import os
+import re
+import requests
 from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
 from openai import OpenAI
@@ -25,6 +27,7 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# Gets FAKE code, meant for the LLM only and is not valid Python
 @app.post("/update_chat")
 async def update_chat(request: Request):
     global latest_blockly_chat_code
@@ -47,6 +50,125 @@ async def set_api_key_chat(request: Request):
     print(f"[CHAT API KEY] Set OPENAI_API_KEY in chat.py environment")
     return {"success": True}
 
+def execute_mcp(mcp_call):
+    """Execute MCP call using the actual Python function from test.py"""
+    global stored_api_key, latest_blockly_chat_code
+
+    if stored_api_key:
+        os.environ["OPENAI_API_KEY"] = stored_api_key
+
+    try:
+        # Now, retrieve the real generated Python code from test.py
+        blockly_code = ""
+        try:
+            resp = requests.get("http://localhost:7860/get_latest_code")
+            if resp.ok:
+                blockly_code = resp.json().get("code", "")
+        except Exception as e:
+            print(f"[WARN] Could not fetch real Python code: {e}")
+
+        if not blockly_code.strip():
+            return "No Python code available from test.py"
+
+        # Parse the MCP call arguments
+        match = re.match(r'create_mcp\((.*)\)', mcp_call.strip())
+        if not match:
+            return "Invalid MCP call format"
+
+        params_str = match.group(1)
+        user_inputs = []
+
+        if params_str:
+            import ast
+            try:
+                dict_str = "{" + params_str.replace("=", ":") + "}"
+                param_dict = ast.literal_eval(dict_str)
+                user_inputs = [str(v) for v in param_dict.values()]
+            except Exception:
+                for pair in params_str.split(','):
+                    if '=' in pair:
+                        _, value = pair.split('=', 1)
+                        user_inputs.append(value.strip().strip('"').strip("'"))
+
+        # Prepare to execute
+        result = ""
+        lines = blockly_code.split('\n')
+        filtered_lines = []
+        skip_mode = False
+        in_demo_block = False
+
+        for line in lines:
+            if 'import gradio' in line:
+                continue
+            if 'demo = gr.Interface' in line:
+                in_demo_block = True
+                skip_mode = True
+                continue
+            elif 'demo.launch' in line:
+                skip_mode = False
+                in_demo_block = False
+                continue
+            elif in_demo_block:
+                continue
+            if 'gr.' in line:
+                continue
+            if not skip_mode:
+                filtered_lines.append(line)
+
+        code_to_run = '\n'.join(filtered_lines)
+
+        def capture_result(msg):
+            nonlocal result
+            result = msg
+
+        env = {
+            "reply": capture_result,
+            "__builtins__": __builtins__,
+        }
+
+        exec("import os", env)
+        exec("import requests", env)
+        exec("import json", env)
+
+        exec(code_to_run, env)
+
+        if "create_mcp" in env:
+            import inspect
+            sig = inspect.signature(env["create_mcp"])
+            params = list(sig.parameters.values())
+
+            typed_args = []
+            for i, arg in enumerate(user_inputs):
+                if i >= len(params):
+                    break
+                if arg is None or arg == "":
+                    typed_args.append(None)
+                    continue
+                anno = params[i].annotation
+                try:
+                    if anno == int:
+                        typed_args.append(int(float(arg)))
+                    elif anno == float:
+                        typed_args.append(float(arg))
+                    elif anno == bool:
+                        typed_args.append(str(arg).lower() in ("true", "1"))
+                    elif anno == str or anno == inspect._empty:
+                        typed_args.append(str(arg))
+                    else:
+                        typed_args.append(arg)
+                except Exception:
+                    typed_args.append(arg)
+
+            result = env["create_mcp"](*typed_args)
+
+        return result if result else "No output generated"
+
+    except Exception as e:
+        print(f"[MCP EXECUTION ERROR] {e}")
+        import traceback
+        traceback.print_exc()
+        return f"Error executing MCP: {str(e)}"
+
 def create_gradio_interface():
     # Hardcoded system prompt
     SYSTEM_PROMPT = """You are an AI assistant created to help with Blockly MCP tasks. Users can create MCP (multi-context-protocol) servers
@@ -64,7 +186,19 @@ for you (the assistant) and is not visible to the user - so do not use this form
 When the user asks questions or talks about their project, don't talk like a robot. This means a few things:
 - Do not say "multi-context-protocol" just say MCP
 - When talking about their project, talk in natural language. Such as if they ask what their project is doing, don't say what the blocks
-are doing, state the goal or things. Remember, that info is just for you and you need to speak normally to the user."""
+are doing, state the goal or things. Remember, that info is just for you and you need to speak normally to the user.
+
+Additionally, you have the ability to use the MCP yourself. Unlike normal OpenAI tools, you call this through chat. To do so, end your msg
+(you cannot say anything after this) with:
+
+```mcp
+create_mcp(input_name=value)
+```
+
+Where you define all the inputs with set values and don't say outputs. And also notice how it doesn't say inputs(). This is just normal
+Python code, not the special syntax.
+
+So, if the user asks you to run the MCP, YOU HAVE THE ABILITY TO. DO NOT SAY THAT YOU CANNOT."""
     
     def chat_with_context(message, history):
         # Check if API key is set and create/update client
@@ -111,7 +245,57 @@ are doing, state the goal or things. Remember, that info is just for you and you
                     {"role": "user", "content": message}
                 ]
             )
-            return response.choices[0].message.content
+            
+            ai_response = response.choices[0].message.content
+            
+            # Check if the response contains ```mcp code block
+            mcp_pattern = r'```mcp\n(.+?)\n```'
+            mcp_match = re.search(mcp_pattern, ai_response, re.DOTALL)
+            
+            if mcp_match:
+                # Extract MCP call
+                mcp_call = mcp_match.group(1)
+                
+                # Filter out the MCP block from the displayed message
+                displayed_response = ai_response[:mcp_match.start()].rstrip()
+                
+                print(f"[MCP DETECTED] Executing: {mcp_call}")
+                
+                # Execute the MCP call
+                mcp_result = execute_mcp(mcp_call)
+                
+                print(f"[MCP RESULT] {mcp_result}")
+                
+                # Add MCP execution to history for context
+                full_history.append({"role": "user", "content": message})
+                full_history.append({"role": "assistant", "content": ai_response})
+                full_history.append({"role": "system", "content": f"MCP execution result: {mcp_result}"})
+                
+                # Call GPT again with the MCP result
+                try:
+                    follow_up_response = client.chat.completions.create(
+                        model="gpt-3.5-turbo",
+                        messages=[
+                            {"role": "system", "content": full_system_prompt},
+                            *full_history,
+                            {"role": "user", "content": "Please respond to the MCP execution result above and provide any relevant information to the user."}
+                        ]
+                    )
+                    
+                    # Combine the filtered initial response with the follow-up
+                    final_response = displayed_response
+                    if displayed_response:
+                        final_response += "\n\n"
+                    final_response += f"**MCP Execution Result:** {mcp_result}\n\n"
+                    final_response += follow_up_response.choices[0].message.content
+                    
+                    return final_response
+                except Exception as e:
+                    return f"{displayed_response}\n\n**MCP Execution Result:** {mcp_result}\n\nError generating follow-up: {str(e)}"
+            
+            # No MCP block found, return normal response
+            return ai_response
+            
         except Exception as e:
             return f"Error: {str(e)}"
     

project/src/index.js

@@ -176,7 +176,7 @@ cancelApiKeyButton.addEventListener("click", () => {
   apiKeyModal.style.display = 'none';
 });
 
-const weatherText = `{"workspaceComments":[{"height":120,"width":479,"id":"XI5[EHp-Ow+kinXf6n5y","x":51.234375,"y":-83,"text":"Gets temperature of location with a latitude and a longitude.\\n\\nThe API requires a minimum of one decimal point to work."}],"blocks":{"languageVersion":0,"blocks":[{"type":"create_mcp","id":")N.HEG1x]Z/,k#TeWr,S","x":50,"y":50,"deletable":false,"extraState":{"inputCount":2,"inputNames":["latitude","longitude"],"inputTypes":["integer","integer"],"outputCount":1,"outputNames":["output0"],"outputTypes":["string"],"toolCount":0},"inputs":{"X0":{"block":{"type":"input_reference_latitude","id":"]3mj!y}qfRt+!okheU7L","deletable":false,"extraState":{"ownerBlockId":")N.HEG1x]Z/,k#TeWr,S"},"fields":{"VARNAME":"latitude"}}},"X1":{"block":{"type":"input_reference_longitude","id":"Do/{HFNGSd.!;POiKS?D","deletable":false,"extraState":{"ownerBlockId":")N.HEG1x]Z/,k#TeWr,S"},"fields":{"VARNAME":"longitude"}}},"R0":{"block":{"type":"in_json","id":"R|j?_8s^H{l0;UZ-oQt3","fields":{"NAME":"temperature_2m"},"inputs":{"JSON":{"block":{"type":"in_json","id":"X=M,R1@7bRjJVZIPi[qD","fields":{"NAME":"current"},"inputs":{"JSON":{"block":{"type":"call_api","id":"^(.vyM.yni08S~c1EBm=","fields":{"METHOD":"GET"},"inputs":{"URL":{"shadow":{"type":"text","id":"}.T;_U_OsRS)B_y09p % { ","fields":{"TEXT":""}},"block":{"type":"text_replace","id":"OwH9uERJPTGQG!UER#ch","inputs":{"FROM":{"shadow":{"type":"text","id":"ya05#^ 7 % UbUeXX#eDSmH","fields":{"TEXT":"{latitude}"}}},"TO":{"shadow":{"type":"text","id":": _ZloQuh9c-MNf-U]!k5","fields":{"TEXT":""}},"block":{"type":"input_reference_latitude","id":"?%@)3sErZ)}=#4ags#gu","extraState":{"ownerBlockId":")N.HEG1x]Z/,k#TeWr,S"},"fields":{"VARNAME":"latitude"}}},"TEXT":{"shadow":{"type":"text","id":"w@zsP)m6:WjkUp,ln3$x","fields":{"TEXT":""}},"block":{"type":"text_replace","id":"ImNPsvzD7r^+1MJ%IirV","inputs":{"FROM":{"shadow":{"type":"text","id":"%o(3rro?WLIFpmE0#MMM","fields":{"TEXT":"{longitude}"}}},"TO":{"shadow":{"type":"text","id":"Zpql-%oJ_sdSi | r |* er | ","fields":{"TEXT":""}},"block":{"type":"input_reference_longitude","id":"WUgiJP$X + zY#f$5nhnTX","extraState":{"ownerBlockId":") N.HEG1x]Z /, k#TeWr, S"},"fields":{"VARNAME":"longitude"}}},"TEXT":{"shadow":{"type":"text","id":", (vw$o_s7P = b4P; 8]}yj","fields":{"TEXT":"https://api.open-meteo.com/v1/forecast?latitude={latitude}&longitude={longitude}&current=temperature_2m,wind_speed_10m"}}}}}}}}}}}}}}}}}}}}]}}`;
+const weatherText = `{"workspaceComments":[{"height":120,"width":479,"id":"XI5[EHp-Ow+kinXf6n5y","x":51.234375,"y":-83,"text":"Gets temperature of location with a latitude and a longitude.\\n\\nThe API requires a minimum of one decimal point to work."}],"blocks":{"languageVersion":0,"blocks":[{"type":"create_mcp","id":")N.HEG1x]Z/,k#TeWr,S","x":50,"y":50,"deletable":false,"extraState":{"inputCount":2,"inputNames":["latitude","longitude"],"inputTypes":["string","string"],"outputCount":1,"outputNames":["output0"],"outputTypes":["string"],"toolCount":0},"inputs":{"X0":{"block":{"type":"input_reference_latitude","id":"]3mj!y}qfRt+!okheU7L","deletable":false,"extraState":{"ownerBlockId":")N.HEG1x]Z/,k#TeWr,S"},"fields":{"VARNAME":"latitude"}}},"X1":{"block":{"type":"input_reference_longitude","id":"Do/{HFNGSd.!;POiKS?D","deletable":false,"extraState":{"ownerBlockId":")N.HEG1x]Z/,k#TeWr,S"},"fields":{"VARNAME":"longitude"}}},"R0":{"block":{"type":"in_json","id":"R|j?_8s^H{l0;UZ-oQt3","fields":{"NAME":"temperature_2m"},"inputs":{"JSON":{"block":{"type":"in_json","id":"X=M,R1@7bRjJVZIPi[qD","fields":{"NAME":"current"},"inputs":{"JSON":{"block":{"type":"call_api","id":"^(.vyM.yni08S~c1EBm=","fields":{"METHOD":"GET"},"inputs":{"URL":{"shadow":{"type":"text","id":"}.T;_U_OsRS)B_y09p % { ","fields":{"TEXT":""}},"block":{"type":"text_replace","id":"OwH9uERJPTGQG!UER#ch","inputs":{"FROM":{"shadow":{"type":"text","id":"ya05#^ 7 % UbUeXX#eDSmH","fields":{"TEXT":"{latitude}"}}},"TO":{"shadow":{"type":"text","id":": _ZloQuh9c-MNf-U]!k5","fields":{"TEXT":""}},"block":{"type":"input_reference_latitude","id":"?%@)3sErZ)}=#4ags#gu","extraState":{"ownerBlockId":")N.HEG1x]Z/,k#TeWr,S"},"fields":{"VARNAME":"latitude"}}},"TEXT":{"shadow":{"type":"text","id":"w@zsP)m6:WjkUp,ln3$x","fields":{"TEXT":""}},"block":{"type":"text_replace","id":"ImNPsvzD7r^+1MJ%IirV","inputs":{"FROM":{"shadow":{"type":"text","id":"%o(3rro?WLIFpmE0#MMM","fields":{"TEXT":"{longitude}"}}},"TO":{"shadow":{"type":"text","id":"Zpql-%oJ_sdSi | r |* er | ","fields":{"TEXT":""}},"block":{"type":"input_reference_longitude","id":"WUgiJP$X + zY#f$5nhnTX","extraState":{"ownerBlockId":") N.HEG1x]Z /, k#TeWr, S"},"fields":{"VARNAME":"longitude"}}},"TEXT":{"shadow":{"type":"text","id":", (vw$o_s7P = b4P; 8]}yj","fields":{"TEXT":"https://api.open-meteo.com/v1/forecast?latitude={latitude}&longitude={longitude}&current=temperature_2m,wind_speed_10m"}}}}}}}}}}}}}}}}}}}}]}}`;
 weatherButton.addEventListener("click", () => {
   try {
     const fileContent = JSON.parse(weatherText);

project/test.py

@@ -19,6 +19,7 @@ latest_blockly_code = ""
 stored_api_key = ""  # Store the API key in memory
 
 
+# Gets REAL Python code, not the LLM DSL
 @app.post("/update_code")
 async def update_code(request: Request):
     global latest_blockly_code
@@ -27,6 +28,13 @@ async def update_code(request: Request):
     print("\n[FASTAPI] Updated Blockly code:\n", latest_blockly_code)
     return {"ok": True}
 
+# Sends the latest code to chat.py so that the agent will be able to use the MCP
+@app.get("/get_latest_code")
+async def get_latest_code():
+    """Return the latest Blockly-generated Python code for other services (like chat.py)"""
+    global latest_blockly_code
+    return {"code": latest_blockly_code}
+
 @app.get("/get_api_key")
 async def get_api_key_endpoint():
     """Get the current API key from memory"""