SimpleChatBot / app.py
thisisdev's picture
Update app.py
8de9df1 verified
import keyfile
import streamlit as st
from langchain_openai import OpenAI
import getpass
import os
import google.generativeai as genai
import pandas as pd
genai.configure(api_key=keyfile.GEMINI_API_KEY)
# 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
model = genai.GenerativeModel("gemini-1.5-flash")
response = model.generate_content(question)
return response.text
# Using Streamlit for generation of page
st.set_page_config(page_title = "ASK GPT / GEMINI", page_icon = ":robot:")
st.header("ASK GPT / GEMINI 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 to the query: ")
st.write(resp)