generate.py

import gradio as gr
import google.generativeai as genai

def generate_text(prompt):
  """
  Generates text using the Gemini model.

  Args:
    prompt: The input prompt for the model.

  Returns:
    The generated text.
  """
  genai.configure(api_key="AIzaSyD6S-CxJyBmyffD8ryHEs75Chj7Y1feh0g") 
  model = genai.GenerativeModel("gemini-2.0-flash-exp")
  response = model.generate_content(prompt)
  return response.text

# Create the Gradio interface
iface = gr.Interface(
    fn=generate_text, 
    inputs="text", 
    outputs="text", 
    title="Gemini Text Generator",
    description="Generate text using the Gemini-2.0-flash-exp model."
)

# Launch the interface
iface.launch()