import google.generativeai as genai
import gradio as gr
from dotenv import load_dotenv
import os

load_dotenv()

api_key = os.getenv("api_key")

model = genai.GenerativeModel('gemini-pro')
genai.configure(api_key = api_key )

def get_response(input_text):
    
    response = model.generate_content(input_text)
    return response.text

iface = gr.Interface(
    fn = get_response,
    inputs = ["text"],
    outputs = ["text"]
)
iface.launch()