SlothioAI commited on
Commit
7295ccf
β€’
1 Parent(s): 4e13ad6

Upload GptWeb3.py

Browse files
Files changed (1) hide show
  1. GptWeb3.py +42 -0
GptWeb3.py ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from langchain.agents import Tool, load_tools, initialize_agent
3
+ from langchain.memory import ConversationBufferMemory
4
+ from langchain.chat_models import ChatOpenAI
5
+ from langchain.utilities import GoogleSearchAPIWrapper
6
+ from langchain.agents import initialize_agent
7
+ from langchain.utilities import WikipediaAPIWrapper
8
+ from langchain.utilities.wolfram_alpha import WolframAlphaAPIWrapper
9
+ import gradio as gr
10
+
11
+ description = """
12
+ πŸ‘ 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/)
13
+ """
14
+
15
+ def initialize_keys(openai_api_key, google_api_key, google_cse_id, wolfram_alpha_appid):
16
+ os.environ['OPENAI_API_KEY'] = openai_api_key
17
+ os.environ['GOOGLE_API_KEY'] = google_api_key
18
+ os.environ['GOOGLE_CSE_ID'] = google_cse_id
19
+ os.environ['WOLFRAM_ALPHA_APPID'] = wolfram_alpha_appid
20
+
21
+ def chat_response(input_text, openai_api_key, google_api_key, google_cse_id, wolfram_alpha_appid):
22
+ initialize_keys(openai_api_key, google_api_key, google_cse_id, wolfram_alpha_appid)
23
+
24
+ llm=ChatOpenAI(temperature=1)
25
+ search = GoogleSearchAPIWrapper()
26
+ tools = load_tools(["google-search", "wolfram-alpha", "wikipedia"], llm=llm)
27
+ memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
28
+
29
+ agent_chain = initialize_agent(tools, llm, agent="zero-shot-react-description",
30
+ verbose=True, memory=memory)
31
+ response = agent_chain.run(input=input_text)
32
+ return response
33
+
34
+ input_text = gr.inputs.Textbox(lines=2, label='Input Text')
35
+ openai_api_key = gr.inputs.Textbox(lines=1, label='OpenAI API Key', type='password')
36
+ google_api_key = gr.inputs.Textbox(lines=1, label='Google API Key', type='password')
37
+ google_cse_id = gr.inputs.Textbox(lines=1, label='Google CSE ID', type='password')
38
+ wolfram_alpha_appid = gr.inputs.Textbox(lines=1, label='Wolfram Alpha AppID', type='password')
39
+
40
+ 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)
41
+
42
+ interface.launch(share=True)