app.py

import base64
import os
from google import genai
from google.genai import types
import gradio as gr

def generate_text(user_input):
    try:
        client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
        
        model = "gemini-2.5-pro-exp-03-25"
        contents = [
            types.Content(
                role="user",
                parts=[
                    types.Part.from_text(text=user_input),
                ],
            ),
        ]
        generate_content_config = types.GenerateContentConfig(
            temperature=2,
            response_mime_type="text/plain",
        )

        response = []
        for chunk in client.models.generate_content_stream(
            model=model,
            contents=contents,
            config=generate_content_config,
        ):
            response.append(chunk.text)
            
        return ''.join(response)
    
    except Exception as e:
        return f"Error: {str(e)}"

iface = gr.Interface(
    fn=generate_text,
    inputs=gr.Textbox(lines=4, placeholder="Enter your prompt here...", label="Input Text"),
    outputs=gr.Textbox(label="Generated Text"),
    title="Gemini 2.5 Pro Text Generator",
    description="Generate text using the Gemini 2.5 Pro model with streaming support",
    allow_flagging="never"
)

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