update

app.py

@@ -11,45 +11,7 @@ import os
 
 from langchain.chains import LLMMathChain, SQLDatabaseChain
 from langchain.agents import Tool, load_tools, initialize_agent, AgentType
-
-def conversation_agent():
-    model = OpenAI(openai_api_key=os.environ['OPENAI_API_KEY'])
-    tools = load_tools(['llm-math', 'terminal'], llm=model)
-
-    prompt = PromptTemplate(template="{question}", input_variables=['question'])
-    llm_chain = LLMChain(llm=model, prompt=prompt)
-    llm_tool = Tool(name="Search", func=llm_chain.run, description="general QA")
-    tools.append(llm_tool)
-
-    memory = ConversationBufferMemory(memory_key="chat_history")
-    conversation_agent = initialize_agent(tools=tools,
-                                          llm=model,
-                                       max_iterations=3,
-                                       verbose=True,
-                                       agent=AgentType.CONVERSATIONAL_REACT_DESCRIPTION,
-                                       memory=memory)
-    resp = conversation_agent("what is (4.5*2.1)^2.2?")
-    print(resp)
-
-    # resp = conversation_agent("if Mary has four apples and Giorgio brings two and a half apple "
-    #             "boxes (apple box contains eight apples), how many apples do we "
-    #             "have?")
-    # print(resp)
-
-    resp = conversation_agent("what is the capital of Norway?")
-    print(resp)
-
-    resp = conversation_agent("what's the most famous landmark of this city")
-    print(resp)
-
-    resp = conversation_agent("free -h")
-    print(resp)
-
-    print("--------")
-    print(conversation_agent.agent.llm_chain.prompt.template)
-
 from langchain.agents.react.base import DocstoreExplorer
-from langchain.docstore import Wikipedia
 
 import gradio as gr 
 
@@ -77,7 +39,10 @@ def demo8():
         print(f"resp: {resp}")
         history.append(resp['output'])
         dial = [(u, v) for u, v in zip(history[::2], history[1::2])]
-        return dial, history
+        return {
+                chatbot: dial,
+                state: history
+            }
 
 
     with gr.Blocks() as demo: