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)