fix: remote exec

app.py

@@ -1,22 +1,11 @@
 import gradio as gr
-from lavague.ActionEngine import ActionEngine
-from lavague.defaults import DefaultLocalLLM, DefaultLLM
-from llama_index.llms.huggingface import HuggingFaceInferenceAPI
 import base64
 import requests
 import uuid
-import re
-
-MAX_CHARS = 1500
 
 USER_ID = str(uuid.uuid4())
 SERVER_URL = "https://lavague.mithrilsecurity.io"
 
-# Use this action_engine instead to have a local inference
-# action_engine = ActionEngine(llm=DefaultLocalLLM())
-
-action_engine = ActionEngine()
-
 title = """
 <div align="center">
   <h1>🌊 Welcome to LaVague</h1>
@@ -26,12 +15,12 @@ title = """
 
 # action_engine = ActionEngine(llm, embedder)
 
-def exec_code_req(url, code):
+def exec_code_req(url, query):
     headers = {
         "X-User-ID": USER_ID  # Include the X-User-ID header for authentication
     }
     try:
-        response = requests.post(SERVER_URL + "/execute_req", json={"url": url, "requests": code}, headers=headers)
+        response = requests.post(SERVER_URL + "/execute_req", json={"url": url, "query": query}, headers=headers)
         if response.status_code == 200:
             return response.json()
         else:
@@ -73,42 +62,21 @@ def process_url(url):
     f.write(scr)
     return "screenshot.png"
 
-def process_instruction(query, url_input):
-    r = get_html(url_input)
-    state = r["html"]
-    query_engine = action_engine.get_query_engine(state)
-    response = query_engine.query(query)
-    source_nodes = response.get_formatted_sources(MAX_CHARS)
-    return response.response, source_nodes
-
-def extract_first_python_code(markdown_text):
-    # Pattern to match the first ```python ``` code block
-    pattern = r"```python(.*?)```"
-    
-    # Using re.DOTALL to make '.' match also newlines
-    match = re.search(pattern, markdown_text, re.DOTALL)
-    if match:
-        # Return the first matched group, which is the code inside the ```python ```
-        return match.group(1).strip()
-    else:
-        # Return None if no match is found
-        return None
-
-
-def exec_code(code, source_nodes, full_code, url):
-    code = extract_first_python_code(code)
+def exec_code(code, source_nodes, full_code, url, query):
     html = ""
     try:
-        r = exec_code_req(url, code)
+        r = exec_code_req(url, query)
         url = r["url"]
         html = r["html"]
+        code = r["code"]
+        source_nodes = r["source_nodes"]
         output = "Successful code execution"
         status = """<p style="color: green; font-size: 20px; font-weight: bold;">Success!</p>"""
         full_code += code
     except Exception as e:
         output = f"Error in code execution: {str(e)}"
         status = """<p style="color: red; font-size: 20px; font-weight: bold;">Failure! Open the Debug tab for more information</p>"""
-    return output, code, html, status, full_code, url
+    return output, code, html, status, full_code, url, source_nodes
 
 def update_image_display(img, url):
     r = send_request(url)
@@ -156,10 +124,8 @@ def create_demo(base_url, instructions):
       # Linking components
       url_input.submit(process_url, inputs=url_input, outputs=image_display, queue=False)
       text_area.submit(show_processing_message, outputs=[status_html], queue=False).then(
-          process_instruction, inputs=[text_area, url_input], outputs=[code_display, source_display], queue=False
-          ).then(
-          exec_code, inputs=[code_display, source_display, full_code, url_input], 
-          outputs=[log_display, code_display, full_html, status_html, full_code, url_input], queue=False
+          exec_code, inputs=[code_display, source_display, full_code, url_input, text_area], 
+          outputs=[log_display, code_display, full_html, status_html, full_code, url_input, source_display], queue=False
       ).then(
           update_image_display, inputs=[image_display, url_input], outputs=[image_display, url_input], queue=False
       )