Update app.py

app.py

@@ -2,9 +2,7 @@ import os
 import gradio as gr
 import pinecone
 from gpt_index import GPTIndexMemory, GPTPineconeIndex
-import streamlit as st
 
-from streamlit_chat import message as st_message
 from langchain.agents import Tool
 from langchain.chains.conversation.memory import ConversationBufferMemory
 from langchain import OpenAI
@@ -15,31 +13,96 @@ PINECONE_API_KEY=os.environ.get("PINECONE_API_KEY")
 PINECONE_ENV=os.environ.get("PINECONE_ENV")
 
 pointing_pinecone_index=pinecone.Index("sethgodin")
-main_index = GPTPineconeIndex([],pinecone_index=pointing_pinecone_index)
-tools = [
-    Tool(
-        name = "GPT Index",
-        func=lambda q: str(main_index.query(q)),
-        description="useful for when you want to answer questions about the author. The input to this tool should be a complete english sentence.",
-        return_direct=True
-    ),
-]
-
-memory = GPTIndexMemory(index=main_index, memory_key="chat_history", query_kwargs={"response_mode": "compact"})
-llm=OpenAI(temperature=0)
-agent_chain = initialize_agent([], llm, agent="conversational-react-description", memory=memory)
-
-def generate_answer():
-    user_message=st.session_state.input_text
-    message_bot = agent_chain.run(input=user_message)
-    st.session_state.history.append({"message": user_message, "is_user": True})
-    st.session_state.history.append({"message": message_bot, "is_user": False})
-    
-st.text_input("Response", key="input_text", on_change=generate_answer)
-
-for chat in st.session_state.history:
-    st_message(**chat)
-st.text_input("Response", key="input_text", on_change=generate_answer)
-
-for chat in st.session_state.history:
-    st_message(**chat)                               
+vectorstore=GPTPineconeIndex([],pinecone_index=pointing_pinecone_index)
+
+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["OPENAI_API_KEY"] = api_key
+        chain = get_chain(vectorstore)
+        os.environ["OPENAI_API_KEY"] = ""
+        return chain
+
+
+class ChatWrapper:
+
+    def __init__(self):
+        self.lock = Lock()
+    def __call__(
+        self, api_key: str, inp: str, history: Optional[Tuple[str, str]], chain
+    ):
+        """Execute the chat functionality."""
+        self.lock.acquire()
+        try:
+            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 OpenAI key to use"))
+                return history, history
+            # Set OpenAI key
+            import openai
+            openai.api_key = api_key
+            # Run chain and append input.
+            output = chain({"question": inp, "chat_history": history})["answer"]
+            history.append((inp, output))
+        except Exception as e:
+            raise e
+        finally:
+            self.lock.release()
+        return history, history
+
+chat = ChatWrapper()
+
+block = gr.Blocks(css=".gradio-container {background-color: lightgray}")
+
+with block:
+    with gr.Row():
+        gr.Markdown("<h3><center>Chat-Your-Data (State-of-the-Union)</center></h3>")
+
+        openai_api_key_textbox = gr.Textbox(
+            placeholder="Paste your OpenAI API key (sk-...)",
+            show_label=False,
+            lines=1,
+            type="password",
+        )
+
+    chatbot = gr.Chatbot()
+
+    with gr.Row():
+        message = gr.Textbox(
+            label="What's your question?",
+            placeholder="Ask questions about the most recent state of the union",
+            lines=1,
+        )
+        submit = gr.Button(value="Send", variant="secondary").style(full_width=False)
+
+    gr.Examples(
+        examples=[
+            "What did the president say about Kentaji Brown Jackson",
+            "Did he mention Stephen Breyer?",
+            "What was his stance on Ukraine",
+        ],
+        inputs=message,
+    )
+
+    gr.HTML("Demo application of a LangChain chain.")
+
+    gr.HTML(
+        "<center>Powered by <a href='https://github.com/hwchase17/langchain'>LangChain 🦜️🔗</a></center>"
+    )
+
+    state = gr.State()
+    agent_state = gr.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])
+
+    openai_api_key_textbox.change(
+        set_openai_api_key,
+        inputs=[openai_api_key_textbox],
+        outputs=[agent_state],
+    )
+
+block.launch(debug=True)