import gradio as gr
from huggingface_hub import InferenceClient
import google.generativeai as genai

# Configure the API key
genai.configure(api_key="AIzaSyBniYJaUYFbfccmg1lkEU_HOa3jIML3Ow4")

# Initialize the model
model = genai.GenerativeModel('gemini-1.5-pro')

def generate_text(prompt):
    response = model.generate_content(prompt)
    return response.text

# ChatInterface expects a function that takes a message and history
def chat_function(message, history):
    generated_text = generate_text(message)
    print("\nGenerated text:")
    print(generated_text)
    return generated_text

demo = gr.ChatInterface(
    fn=chat_function,
    title="Gemini-1.5-Pro Chat Interface",
    description="Enter your prompt to generate text using Gemini-1.5-Pro"
)

if __name__ == "__main__":
    demo.launch()