Update app.py

app.py

@@ -3,7 +3,6 @@ import os
 import subprocess
 import random
 from huggingface_hub import InferenceClient
-import gradio as gr
 from safe_search import safe_search
 from i_search import google
 from i_search import i_search as i_s
@@ -44,7 +43,7 @@ logger = logging.getLogger(__name__)
 # Load custom prompts
 try:
     with open('custom_prompts.yaml', 'r') as fp:
-        custom_prompts = yaml.load(fp, Loader=yaml.FullLoader)
+        custom_prompts = yaml.safe_load(fp)
 except FileNotFoundError:
     custom_prompts = {
         "WEB_DEV": "",
@@ -285,7 +284,7 @@ def run_action(purpose, task, history, directory, action_name, action_input):
                 print("COMPRESSING HISTORY")
             history = compress_history(purpose, task, history, directory)
         
-        if not action_name in NAME_TO_FUNC:
+        if action_name not in NAME_TO_FUNC:
             action_name = "MAIN"
         if action_name == "" or action_name is None:
             action_name = "MAIN"
@@ -327,10 +326,10 @@ def run(purpose, history):
             action_input,
         )
         
-        yield (history)
+        yield history
         
         if task == "END":
-            return (history)
+            return history
 
 ################################################
 
@@ -344,14 +343,14 @@ def generate(
         prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.7,
 ):
     seed = random.randint(1, 1111111111111111)
-    agent = prompts["WEB_DEV"]
+    agent = custom_prompts["WEB_DEV"]
 
     if agent_name == "WEB_DEV":
-        agent = prompts["WEB_DEV"]
+        agent = custom_prompts["WEB_DEV"]
     elif agent_name == "AI_SYSTEM_PROMPT":
-        agent = prompts["AI_SYSTEM_PROMPT"]
+        agent = custom_prompts["AI_SYSTEM_PROMPT"]
     elif agent_name == "PYTHON_CODE_DEV":
-        agent = prompts["PYTHON_CODE_DEV"]
+        agent = custom_prompts["PYTHON_CODE_DEV"]
     
     system_prompt = agent
     temperature = float(temperature)
@@ -389,15 +388,15 @@ with gr.Blocks() as iface:
 
     iface.load(inputs, events)
     def log_messages(inputs, outputs):
-        logger(f'Input: {inputs}, Output: {outputs}')
+        logger.info(f'Input: {inputs}, Output: {outputs}')
         
 @gr.Interface.load(inputs, agents)
 def log_messages(inputs, outputs):
-    logger(f'Input: {inputs}, Output: {outputs}')
+    logger.info(f'Input: {inputs}, Output: {outputs}')
 
 @gr.Interface.load(inputs, events)
 def log_messages(inputs, outputs):
-    logger(f'Input: {inputs}, Output: {outputs}')
+    logger.info(f'Input: {inputs}, Output: {outputs}')
 
 def update_sys_prompt(agent):
     global SYSTEM_PROMPT
@@ -514,9 +513,10 @@ with iface:
                [msg, chatbot, agent_dropdown, sys_prompt, temperature, max_new_tokens, top_p, repetition_penalty], 
                [chatbot, msg])
     clear.click(lambda: None, None, chatbot, queue=False)
-    gr.Examples(examples, [msg, agent_dropdown, sys_prompt, temperature, max_new_tokens, top_p, repetition_penalty]),
 
-iface.load(fn=get_helpful_tip, inputs=agent_dropdown, outputs=ui.info),
-iface.load(fn=update_sys_prompt, inputs=agent_dropdown, outputs=sys_prompt),
-iface.load(fn=log_messages, input_type= state, event_type= save_model)
+    gr.Examples(examples, [msg, agent_dropdown, sys_prompt, temperature, max_new_tokens, top_p, repetition_penalty])
+
+    agent_dropdown.change(fn=get_helpful_tip, inputs=agent_dropdown, outputs=gr.components.Markdown())
+    agent_dropdown.change(fn=update_sys_prompt, inputs=agent_dropdown, outputs=sys_prompt)
+
 iface.launch()