Update app.py

app.py

@@ -28,15 +28,12 @@ client = InferenceClient(
     "mistralai/Mixtral-8x7B-Instruct-v0.1"
 )
 
-
 ############################################
 
-
 VERBOSE = True
 MAX_HISTORY = 100
 #MODEL = "gpt-3.5-turbo"  # "gpt-4"
 
-
 def format_prompt(message, history):
   prompt = "<s>"
   for user_prompt, bot_response in history:
@@ -45,8 +42,6 @@ def format_prompt(message, history):
   prompt += f"[INST] {message} [/INST]"
   return prompt
 
-
-
 def run_gpt(
     prompt_template,
     stop_tokens,
@@ -87,7 +82,6 @@ def run_gpt(
         print(LOG_RESPONSE.format(resp))
     return resp
 
-
 def compress_history(purpose, task, history, directory):
     resp = run_gpt(
         COMPRESS_HISTORY_PROMPT,
@@ -155,7 +149,6 @@ def call_main(purpose, task, history, directory, action_input):
             #assert False, "unknown action: {}".format(line)
     return "MAIN", None, history, task
 
-
 def call_set_task(purpose, task, history, directory, action_input):
     task = run_gpt(
         TASK_PROMPT,
@@ -243,8 +236,6 @@ def run(purpose,history):
             return (history)
             #return ("", [(purpose,history)])
 
-
-
 ################################################
 
 def format_prompt(message, history):
@@ -295,7 +286,6 @@ def generate(
         yield output
     return output
 
-
 additional_inputs=[
     gr.Dropdown(
         label="Agents",
@@ -346,7 +336,6 @@ additional_inputs=[
         info="Penalize repeated tokens",
     ),
 
-
 ]
 
 examples=[["What are the biggest news stories today?", None, None, None, None, None, ],
@@ -359,32 +348,22 @@ examples=[["What are the biggest news stories today?", None, None, None, None, N
           ["What are some unique features of Rust that make it stand out compared to other systems programming languages like C++?", None, None, None, None, None,],
          ]
 
-'''
-gr.ChatInterface(
-fn=run,
-chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
-title="Mixtral 46.7B\nMicro-Agent\nInternet Search <br> development test",
-examples=examples,
-concurrency_limit=20,
-with gr.Blocks() as ifacea:
-    gr.HTML("""TEST""")
-ifacea.launch()
-).launch()
+def process_input(user_input, history, chatbot):
+    """Processes user input and updates the chatbot."""
+    purpose = "General"
+    history = history + [(user_input, "")]
+    chatbot.append(user_input, "")
+
+    for response in run(purpose, history):
+        chatbot.append("", response)
+        yield chatbot
+
 with gr.Blocks() as iface:
-    #chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
-    chatbot=gr.Chatbot()
-    msg = gr.Textbox()
+    chatbot = gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, layout="panel")
+    msg = gr.Textbox(label="Enter your message")
     with gr.Row():
-        submit_b = gr.Button()
+        submit_b = gr.Button("Submit")
         clear = gr.ClearButton([msg, chatbot])
-    submit_b.click(run, [msg,chatbot],[msg,chatbot])
-    msg.submit(run, [msg, chatbot], [msg, chatbot])
-iface.launch()
-'''
-gr.ChatInterface(
-    fn=run,
-    chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, layout="panel"),
-    title="Mixtral 46.7B\nMicro-Agent\nInternet Search <br> development test",
-    examples=examples,
-    concurrency_limit=50,
-).launch(show_api=True)
+    submit_b.click(process_input, [msg, chatbot, chatbot], chatbot)
+    msg.submit(process_input, [msg, chatbot, chatbot], chatbot)
+iface.launch()