app.py

import keyfile 

import streamlit as st

from langchain_openai import OpenAI

import getpass

import os
 
# Creating a function for getting the responses from OpenAI

def get_response(question):

    os.environ["OPENAI_API_KEY"] = keyfile.OPENAI_API_KEY

    llm = OpenAI()

    answer = llm.invoke(question)

    return answer
 
# Using Streamlit for generation of page

st.set_page_config(page_title = "ASK GPT", page_icon = ":robot:")

st.header("ASK GPT Application")
 
# Create a function for taking user input

def get_user_input():

    text = st.text_input("Ask: ", key = "input")

    return text
 
user_input = get_user_input()

resp = get_response(user_input)
 
# Submission button

submit = st.button("Ask!")
 
if submit:

    st.subheader("Answer: ")

    st.write(resp)