Update app.py

app.py

@@ -28,6 +28,15 @@ client = InferenceClient(
     "mistralai/Mixtral-8x7B-Instruct-v0.1"
 )
 
+
+from gradio_client import Client
+
+client = Client("https://omnibus-html-image-current-tab.hf.space/--replicas/o0seq/")
+
+def get_screenshot(chat,height=5000,width=600,chatblock=[1],header=True,theme="light",wait=3000)
+    result = client.predict(chat,height,width,chatblock,header,theme,wait,api_name="/run_script")
+    print (result)
+
 def parse_action(string: str):
     print("PARSING:")
     print(string)
@@ -366,14 +375,17 @@ examples =[
     "Teach me to fly a helicopter"
 ]
 
+with gr.Blocks() as app:
+    gr.ChatInterface(
+        fn=run,
+        chatbot=gr.Chatbot(show_label=False, show_share_button=True, show_copy_button=True, likeable=True, layout="panel", height="800px"),
+        title="Mixtral 46.7B Powered <br> Search",
+        examples=examples,
+        concurrency_limit=20,
+    )
+    btn=gr.Button()
+app.launch()
 
-gr.ChatInterface(
-    fn=run,
-    chatbot=gr.Chatbot(show_label=False, show_share_button=True, show_copy_button=True, likeable=True, layout="panel", height="800px"),
-    title="Mixtral 46.7B Powered <br> Search",
-    examples=examples,
-    concurrency_limit=20,
-).launch()
 '''
 with gr.Blocks() as app:
     with gr.Row():