Spaces:
Sleeping
Sleeping
File size: 2,232 Bytes
000471a 1b5c226 000471a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
import os
from langchain.agents import Tool, load_tools, initialize_agent
from langchain.memory import ConversationBufferMemory
from langchain.chat_models import ChatOpenAI
from langchain.utilities import GoogleSearchAPIWrapper
from langchain.agents import initialize_agent
from langchain.utilities import WikipediaAPIWrapper
from langchain.utilities.wolfram_alpha import WolframAlphaAPIWrapper
import gradio as gr
description = """
👍 If you liked the app, sign up for my AI newsletter where I'll be sharing prompts, news, and tools like this one: [https://slothio.com/ai-newsletter/](https://slothio.com/ai-newsletter/)
"""
def initialize_keys(openai_api_key, google_api_key, google_cse_id, wolfram_alpha_appid):
os.environ['OPENAI_API_KEY'] = openai_api_key
os.environ['GOOGLE_API_KEY'] = google_api_key
os.environ['GOOGLE_CSE_ID'] = google_cse_id
os.environ['WOLFRAM_ALPHA_APPID'] = wolfram_alpha_appid
def chat_response(input_text, openai_api_key, google_api_key, google_cse_id, wolfram_alpha_appid):
initialize_keys(openai_api_key, google_api_key, google_cse_id, wolfram_alpha_appid)
llm=ChatOpenAI(temperature=1, max_tokens=2000,model_name="gpt-3.5-turbo" )
search = GoogleSearchAPIWrapper()
tools = load_tools(["google-search", "wolfram-alpha", "wikipedia"], llm=llm)
memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
agent_chain = initialize_agent(tools, llm, agent="zero-shot-react-description",
verbose=True, memory=memory)
response = agent_chain.run(input=input_text)
return response
input_text = gr.inputs.Textbox(lines=2, label='Input Text')
openai_api_key = gr.inputs.Textbox(lines=1, label='OpenAI API Key', type='password')
google_api_key = gr.inputs.Textbox(lines=1, label='Google API Key', type='password')
google_cse_id = gr.inputs.Textbox(lines=1, label='Google CSE ID', type='password')
wolfram_alpha_appid = gr.inputs.Textbox(lines=1, label='Wolfram Alpha AppID', type='password')
interface = gr.Interface(fn=chat_response, inputs=[input_text, openai_api_key, google_api_key, google_cse_id, wolfram_alpha_appid], outputs="text", description=description)
interface.launch(share=False) |