Gradio bot for QA on apis

app.py

@@ -0,0 +1,43 @@
+import gradio as gr
+from main import chat, index_url
+
+"""Use text to call chat method from main.py"""
+def add_text(history, text):
+    response = str(chat(question=text))
+    history = history + [(text, response)]
+    return history, ""
+
+"""Use text to call index_url method from main.py"""
+def add_url(history, url):
+    data = index_url(url)
+    response = "Found your data!"
+    history = history + [(url, response)]
+    return history, ""
+
+def bot(history):
+    return history
+
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot([], elem_id="chatbot").style(height=750)
+
+    with gr.Row():
+        with gr.Column(scale=0.65):
+            txt = gr.Textbox(
+                show_label=False,
+                placeholder="Enter text and press enter, or upload an image",
+            ).style(container=False)
+        with gr.Column(scale=0.35, min_width=0):
+            url = gr.Textbox(
+                show_label=False,
+                placeholder="Enter url api endpoint",
+            ).style(container=False)
+
+    txt.submit(add_text, [chatbot, txt], [chatbot, txt]).then(
+        bot, chatbot, chatbot
+    )
+    url.submit(add_url, [chatbot, url], [chatbot, url]).then(
+        bot, chatbot, chatbot
+    )
+
+if __name__ == "__main__":
+    demo.launch()

main.py

@@ -0,0 +1,47 @@
+import os
+
+import requests
+import yaml
+from langchain.agents import (
+    create_json_agent
+)
+from langchain.agents.agent_toolkits import JsonToolkit
+from langchain.chat_models import ChatOpenAI
+from langchain.tools.json.tool import JsonSpec
+
+os.environ['OPENAI_API_KEY'] = "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"
+
+llm = ChatOpenAI(model_name="gpt-4", temperature=0.1)
+agent = None
+
+"""Call the url and use json response to create a yaml file"""
+def index_url(url):
+    global agent
+    response = requests.get(url)
+    json_response = response.json()
+
+    # create yaml from json response
+    yaml_string = yaml.dump(json_response)
+
+    data = yaml.load(yaml_string, Loader=yaml.FullLoader)
+    json_spec = JsonSpec(dict_=data, max_value_length=4000)
+    json_toolkit = JsonToolkit(spec=json_spec)
+    agent = create_json_agent(
+        llm=llm,
+        toolkit=json_toolkit,
+        verbose=True
+    )
+    return data
+
+"""Use the agent to call method run using the text as input"""
+def chat(question):
+    global agent
+    if agent is not None:
+        response = agent.run(question)
+        print(response)
+    else:
+        response = "Please add a request url first"
+    return response
+
+
+

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+langchain
+openai