File size: 1,351 Bytes
5c5a1ae 3fd0067 5c5a1ae d375a16 baface3 d375a16 3fd0067 d375a16 5c5a1ae 5fced44 5c5a1ae 5fced44 d375a16 5c5a1ae d375a16 9b324d1 d375a16 5c5a1ae d375a16 bdfd7a5 5f3d5cb d375a16 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
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() |