Update app.py

app.py

@@ -58,61 +58,6 @@ def format_prompt(message, history):
   prompt += f"[INST] {message} [/INST]"
   return prompt
 
-def call_search(purpose, task, history, action_input):
-    return_list=[]
-    print (action_input)
-    #if action_input in query.tasks:
-    print ("trying")        
-    try:
-        if action_input != "" and action_input != None:
-            action_input.strip('""')
-            #model_list = api.list_models(filter=f"{action_input}",sort="last_modified",limit=1000,direction=-1)
-            #model_list = api.list_models(filter=f"{action_input}",limit=1000)
-            model_list = api.list_models(filter=f"{action_input}")
-            this_obj = list(model_list)
-            print(f'THIS_OBJ :: {this_obj[0]}')
-            for i,eb in enumerate(this_obj):
-                #return_list.append(this_obj[i].id)
-                return_list.append({"id":this_obj[i].id,
-                                    "author":this_obj[i].author,
-                                    "created_at":this_obj[i].created_at,
-                                    "last_modified":this_obj[i].last_modified,
-                                    "private":this_obj[i].private,
-                                    "gated":this_obj[i].gated,
-                                    "disabled":this_obj[i].disabled,
-                                    "downloads":this_obj[i].downloads,
-                                    "likes":this_obj[i].likes,
-                                    "library_name":this_obj[i].library_name,
-                                    "tags":this_obj[i].tags,
-                                    "pipeline_tag":this_obj[i].pipeline_tag,
-                                   })
-            #print (return_list)
-            c=0
-            rl = len(return_list)
-            print(rl)
-            for i in str(return_list):
-                if i == " " or i==",":
-                    c +=1
-            
-            print (c)
-            if rl > MAX_DATA:
-                print("compressing...")
-                return_list = compress_data(rl,purpose,task,return_list)
-            history = "observation: the search results are:\n {}\n".format(return_list)
-            return "MAIN", None, history, task
-        else: 
-            history = "observation: I need to trigger a search using the following syntax:\naction: SEARCH action_input=URL\n"
-            return "UPDATE-TASK", None, history, task
-    except Exception as e:
-        print (e)
-        history = "observation: I need to trigger a search using the following syntax:\naction: SEARCH action_input=URL\n"
-        return "UPDATE-TASK", None, history, task
-
-        #else:
-    #    history = "observation: The search query I used did not return a valid response"
-        
-    return "MAIN", None, history, task
-
 
 def run_gpt(
     prompt_template,
@@ -422,7 +367,7 @@ def run(purpose,history):
 
 examples =[
     "What is the current weather in Florida?",
-    "what are todays breaking news stories?",
+    "Find breaking news about Texas",
     "Find the best deals on flippers for scuba diving",
     "Teach me to fly a helicopter"
 ]
@@ -430,11 +375,10 @@ examples =[
 
 gr.ChatInterface(
     fn=run,
-    chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
+    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,
-    css="height:600px;"
 ).launch()
 '''
 with gr.Blocks() as app:
@@ -457,5 +401,4 @@ with gr.Blocks() as app:
     find_btn.click(find_it,[inp,q,num],[outp,outl])
     
 app.launch()
-
 '''