Update app.py

app.py

@@ -3,7 +3,7 @@ from typing import Optional, Tuple
 
 import gradio as gr
 from langchain.agents import create_pandas_dataframe_agent
-from langchain.llms import Anthropic
+from langchain.llms import OpenAI
 import pandas as pd
 from threading import Lock
 
@@ -13,20 +13,20 @@ def load_data():
 
 def load_chain():
     """Logic for loading the chain you want to use should go here."""
-    llm = Anthropic(temperature=0)
-    trans = load_data()
-    chain = create_pandas_dataframe_agent(llm, trans)
+    llm = OpenAI(temperature=0)
+    chain = create_pandas_dataframe_agent(llm, load_data())
     return chain
 
-def set_anthropic_api_key(api_key: str):
+
+def set_openai_api_key(api_key: str):
     """Set the api key and return chain.
 
     If no api_key, then None is returned.
     """
     if api_key:
-        os.environ["ANTHROPIC_API_KEY"] = api_key
+        os.environ["OPENAI_API_KEY"] = api_key
         chain = load_chain()
-        os.environ["ANTHROPIC_API_KEY"] = ""
+        os.environ["OPENAI_API_KEY"] = ""
         return chain
 
 class ChatWrapper:
@@ -42,11 +42,11 @@ class ChatWrapper:
             history = history or []
             # If chain is None, that is because no API key was provided.
             if chain is None:
-                history.append((inp, "Please paste your anthropic key to use"))
+                history.append((inp, "Please paste your OpenAI key to use"))
                 return history, history
-            # Set anthropic key
-            import anthropic
-            anthropic.api_key = api_key
+            # Set OpenAI key
+            import openai
+            openai.api_key = api_key
             # Run chain and append input.
             output = chain.run(input=inp)
             history.append((inp, output))
@@ -64,8 +64,8 @@ with block:
     with gr.Row():
         gr.Markdown("<h3><center>LangChain Demo</center></h3>")
 
-        anthropic_api_key_textbox = gr.Textbox(
-            placeholder="Paste your anthropic API key (sk-...)",
+        openai_api_key_textbox = gr.Textbox(
+            placeholder="Paste your OpenAI API key (sk-...)",
             show_label=False,
             lines=1,
             type="password",
@@ -76,16 +76,16 @@ with block:
     with gr.Row():
         message = gr.Textbox(
             label="What's your question?",
-            placeholder="What was the average gas price on 2019-01-22?",
+            placeholder="What's the answer to life, the universe, and everything?",
             lines=1,
         )
         submit = gr.Button(value="Send", variant="secondary").style(full_width=False)
 
     gr.Examples(
         examples=[
-            "Which to_address spent the most gas?",
-            "What was the average gas price on 2019-01-22?",
-            "How many unique addresses were sending transactions on 2019-01-22?",
+            "Hi! How's it going?",
+            "What should I do tonight?",
+            "Whats 2 + 2?",
         ],
         inputs=message,
     )
@@ -99,12 +99,12 @@ with block:
     state = gr.State()
     agent_state = gr.State()
 
-    submit.click(chat, inputs=[anthropic_api_key_textbox, message, state, agent_state], outputs=[chatbot, state])
-    message.submit(chat, inputs=[anthropic_api_key_textbox, message, state, agent_state], outputs=[chatbot, state])
+    submit.click(chat, inputs=[openai_api_key_textbox, message, state, agent_state], outputs=[chatbot, state])
+    message.submit(chat, inputs=[openai_api_key_textbox, message, state, agent_state], outputs=[chatbot, state])
 
-    anthropic_api_key_textbox.change(
-        set_anthropic_api_key,
-        inputs=[anthropic_api_key_textbox],
+    openai_api_key_textbox.change(
+        set_openai_api_key,
+        inputs=[openai_api_key_textbox],
         outputs=[agent_state],
     )