pgleeson commited on
Commit
a249ba6
·
verified ·
1 Parent(s): 71a96bf

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +71 -0
app.py ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit
2
+
3
+
4
+ import streamlit as st
5
+ from langchain.llms import OpenAI
6
+
7
+ from langchain.prompts import PromptTemplate
8
+ from langchain.chains import LLMChain
9
+
10
+ LLM_GPT35 = 'GPT3.5'
11
+ LLM_GPT4 = 'GPT4'
12
+
13
+ st.title("Project Sydney")
14
+
15
+ openai_api_key = None
16
+
17
+ with st.sidebar:
18
+ openai_api_key_sb = st.text_input("OpenAI API Key", type="password")
19
+ "[Get an OpenAI API key](https://platform.openai.com/account/api-keys)"
20
+
21
+
22
+ def get_openai_api_key():
23
+
24
+ if openai_api_key_sb == None or len(openai_api_key_sb)==0:
25
+ openai_api_key = str(open('../oaik','r').readline())
26
+ else:
27
+ openai_api_key = openai_api_key_sb
28
+
29
+ return openai_api_key
30
+
31
+
32
+ def generate_response(input_text, llm_ver, temperature):
33
+
34
+ template = """Question: {question}
35
+
36
+ Answer: """
37
+ prompt = PromptTemplate(
38
+ template=template,
39
+ input_variables=['question']
40
+ )
41
+
42
+
43
+ if llm_ver==LLM_GPT35:
44
+ llm = OpenAI(temperature=temperature, openai_api_key=get_openai_api_key())
45
+ elif llm_ver==LLM_GPT4:
46
+
47
+ from langchain_openai import ChatOpenAI
48
+ llm = ChatOpenAI(model_name='gpt-4', openai_api_key=get_openai_api_key())
49
+
50
+ llm_chain = LLMChain(
51
+ prompt=prompt,
52
+ llm=llm
53
+ )
54
+
55
+ response = llm_chain.run(input_text)
56
+
57
+ st.info(response)
58
+
59
+
60
+ with st.form("my_form"):
61
+ text = st.text_area("Enter text:", "What is the primary role of the C. elegans neuron ADAL?")
62
+ llm_ver = st.selectbox('Which LLM version?', (LLM_GPT35, LLM_GPT4))
63
+
64
+ temperature = st.text_input("Temperature", value=0.1)
65
+
66
+ submitted = st.form_submit_button("Submit")
67
+
68
+ if not get_openai_api_key():
69
+ st.info("Please add your OpenAI API key to continue.")
70
+ elif submitted:
71
+ generate_response(text, llm_ver, temperature)