Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import subprocess
 import random
+import torch
 from transformers import pipeline
 import gradio as gr
 from safe_search import safe_search
@@ -37,7 +38,7 @@ def run_gpt(
     **prompt_kwargs,
 ):
     seed = random.randint(1,1111111111111111)
-    print (seed)
+    print(seed)
     generate_kwargs = dict(
         temperature=1.0,
         max_new_tokens=2096,
@@ -90,7 +91,7 @@ def call_search(purpose, task, history, directory, action_input):
             print(response)
             history += "observation: search result is: {}\n".format(response)
         else:
-            history += "observation: I need to provide a valid URL to 'action: SEARCH action_input = {}\n".format(action_input)
+            history += "observation: I need to provide a valid URL to 'action: SEARCH'\n"
     except Exception as e:
         history += "observation: {}'\n".format(e)
     return "MAIN", None, history, task
@@ -132,7 +133,8 @@ def call_main(purpose, task, history, directory, action_input):
 
 
 def call_set_task(purpose, task, history, directory, action_input):
-    task = run_gpt(
+    task = "COMPLETE"
+    resp = run_gpt(
         TASK_PROMPT,
         stop_tokens=[],
         max_tokens=64,
@@ -159,7 +161,7 @@ def run_action(purpose, task, history, directory, action_name, action_input):
     print(f'action_name::{action_name}')
     try:
         if "RESPONSE" in action_name or "COMPLETE" in action_name:
-            action_name = action_name
+            action_name = "COMPLETE"
             task = "COMPLETE"
             return action_name, "COMPLETE", history, task
 
@@ -167,11 +169,11 @@ def run_action(purpose, task, history, directory, action_name, action_input):
         if len(history.split("\n")) > 5:
             if True:
                 print("COMPRESSING HISTORY")
-            history = compress_history(purpose, task, history, directory)
+            history = history
         if not action_name in NAME_TO_FUNC:
-            action_name = action_name
-        if action_name == "MAIN":
-            action_name = action_name
+            action_name = "MAIN"
+        if action_name == "SEARCH":
+            action_name = "SEARCH"
         assert action_name in NAME_TO_FUNC
 
         print("RUN: ", action_name, action_input)
@@ -185,15 +187,15 @@ def run(purpose,history):
 
     #print(purpose)
     #print(hist)
-    task = "MAIN"
-    directory = "MAIN"
+    task = "COMPLETE"
+    directory = "directory"
     if history:
         history = history
     if not history:
         history = ""
 
     action_name = "MAIN"
-    action_input = ""
+    action_input = "input"
     while True:
         print("")
         print("")
@@ -307,8 +309,6 @@ additional_inputs=[
         interactive=True,
         info="Penalize repeated tokens",
     ),
-
-
 ]
 
 examples=[["What are the biggest news stories today?", None, None, None, None, None, ],
@@ -321,10 +321,16 @@ 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,
-).launch(show_api=False)
+import transformers
+import torch
+
+def launch_interface():
+    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,
+    ).launch(show_api=False)
+
+launch_interface()