Added temperature

app.py

@@ -26,6 +26,8 @@ pinecone_index = "group-1"
 
 if 'exchanges' not in st.session_state:
     st.session_state.exchanges = []
+if 'temperature' not in st.session_state:
+    st.session_state.temperature = 0.5
 
 # -------------------- Functions -----------------------
 def console_log(msg):
@@ -56,12 +58,12 @@ def load_pdf(url):
     console_log(f'After splitting, you have {len(texts)} documents')
     load_vector_database()
 
-def chat(query):
+def chat(query, temperature):
 
     from langchain.llms import OpenAI
     from langchain.chains.question_answering import load_qa_chain
 
-    llm = OpenAI(temperature=0, openai_api_key=st.secrets["OPENAI_API_KEY"])
+    llm = OpenAI(temperature=temperature, openai_api_key=st.secrets["OPENAI_API_KEY"])
     chain = load_qa_chain(llm, chain_type="stuff")
    
     embeddings = OpenAIEmbeddings(openai_api_key=st.secrets["OPENAI_API_KEY"])
@@ -109,12 +111,13 @@ with st.sidebar:
     st.button('Click to start loading PDF', key="load_pdf", on_click=load_pdf, args=[pdf_dict[option]])
 
 # ------------------------ Chatbot ------------------------
+st.slider("Temperature (0 = Most Deterministic)", min_value=0.0, max_value=1.0, step=0.1, key="temperature")
 st.text_input("Prompt", placeholder="Ask me anything", key="prompt")
 
 if st.session_state.prompt:
     st.session_state.exchanges.append({"role": "user", "content": st.session_state.prompt})
     try:
-        response = chat(format_prompt(st.session_state.exchanges))
+        response = chat(format_prompt(st.session_state.exchanges), st.session_state.temperature)
     except Exception as e:
         st.error(e) 
         st.stop()