Update app.py

app.py

@@ -1,4 +1,6 @@
-from gradio_client import Client, InferenceClient
+from gradio_client import Client
+from huggingface_hub import InferenceClient
+
 import gradio as gr
 ss_client = Client("https://omnibus-html-image-current-tab.hf.space/")
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
@@ -50,13 +52,19 @@ chat=[('user','bot'),('user','bot')]
 #get_screenshot(chat=[('user','bot'),('user','bot')])
 with gr.Blocks() as app:
     with gr.Row():
-        with gr.Column(scale=2):
+        with gr.Column(scale=3):
             chat_b = gr.Chatbot()
+            with gr.Row():
+                with gr.Column(scale=3):
+                    inp = gr.Textbox(label="Prompt")
+                with gr.Column(scale=1):
+                    btn = gr.Button("Chat")
+                    with gr.Group():
+                        stop_btn=gr.Button("Stop")
+                        clear_btn=gr.Button("Clear")
         with gr.Column(scale=1):
             with gr.Group():    
-                inp = gr.Textbox(label="Prompt")
                 sys_inp = gr.Textbox(label="System Prompt")
-                btn = gr.Button("Chat")
                 im_btn=gr.Button("Screenshot")
                 img=gr.Image(type='filepath')
     btn.click(chat_inf,[sys_inp,inp,chatbot],chatbot)