Upload 2 files

apikey.py

@@ -0,0 +1 @@
+apikey= ''

app.py

@@ -0,0 +1,62 @@
+ # Bring in deps
+import os
+from apikey import apikey
+
+import streamlit as st
+from langchain.llms import OpenAI
+from langchain.prompts import PromptTemplate
+from langchain.chains import LLMChain
+from langchain.memory import ConversationBufferMemory
+from langchain.utilities import WikipediaAPIWrapper
+
+# Set OpenAI API key
+os.environ['OPENAI_API_KEY'] = apikey
+
+# App framework
+st.title('Test bot')
+prompt = st.text_input('Plug in the topic here')
+
+# Check if there's a prompt
+if prompt:
+   
+
+    question_template = PromptTemplate(
+            input_variables=['topic', 'wikipedia_research','answer'],
+            template='create 5 multiple-choice question and answer with the following information: QUESTION: {topic} OPTIONS: {wikipedia_research} ANSWER: {wikipedia_research}'
+    )
+
+
+
+    # Memory
+    
+    question_memory = ConversationBufferMemory(input_key='topic', memory_key='chat_history')
+    
+
+    # Llms
+    llm = OpenAI(temperature=0.9)
+  
+    question_chain = LLMChain(llm=llm, prompt=question_template, verbose=True, output_key='question', memory=question_memory)
+   
+    wiki = WikipediaAPIWrapper()
+
+    # Run chains
+   
+    
+    wiki_research = wiki.run(prompt)
+    question = question_chain.run(topic=prompt, wikipedia_research=wiki_research)
+    
+
+    
+    st.write("Question:", question)
+  
+    
+
+
+    
+    with st.expander('Question history'):
+        st.info(question_memory.buffer)
+
+    with st.expander('Wikipedia Research'):
+        st.info(wiki_research)
+else:
+    st.warning("Please enter your topic.")