Create app.py

app.py

@@ -0,0 +1,53 @@
+import streamlit as st
+from langchain.llms import OpenAI
+from langchain.chains import PALChain
+from langchain.chains.llm import LLMChain
+
+# set page config
+st.set_page_config(page_title="🦜🔗 This is a Program Aided Language model")
+st.title("🦜🔗🧮 Program Aided Language Model: Helps with math problems")
+st.markdown(
+    """This example was adapt from Data Professor Github [repo](https://github.com/dataprofessor/langchain-quickstart/blob/master/streamlit_app.py)"""
+)
+st.markdown(
+    """Paper: [Program-Aided Language Models for Program Synthesis](https://arxiv.org/pdf/2211.10435.pdf)"""
+)
+st.markdown(
+    """Credit: [Sam Witteven](https://www.youtube.com/playlist?list=PL8motc6AQftk1Bs42EW45kwYbyJ4jOdiZ)"""
+)
+
+# sidebar for OpenAI API key
+openai_api_key = st.sidebar.text_input("Enter your OpenAI API key", type="password")
+
+
+def generate_response(input_text):
+    """
+    Generates response to input text using PALChain
+
+    Parameters
+    ----------
+    input_text : str
+      Input text to generate response for using PALChain
+
+    Returns
+    -------
+    None
+
+    Example
+    -------
+    """
+    llm = OpenAI(temperature=0, openai_api_key=openai_api_key, max_tokens=512)
+    pal_chain = PALChain.from_math_prompt(llm, verbose=True)
+    st.write(pal_chain.run(input_text))
+
+
+with st.form("my_form"):
+    text = st.text_area(
+        "Enter text:",
+        "Leah had 32 chocolates and her sister had 42. If they ate 35, how many pieces do they have left in total?",
+    )
+    submitted = st.form_submit_button("Submit")
+    if not openai_api_key.startswith("sk-"):
+        st.warning("Please enter your OpenAI API key!", icon="⚠")
+    if submitted and openai_api_key.startswith("sk-"):
+        generate_response(text)