app.py

import base64
import os
from google import genai
from google.genai import types
import gradio as gr
from IPython.display import display
from IPython.display import Markdown

client = genai.Client(
    api_key=os.environ.get("GEMINI_API_KEY"),
)

def generate(prompt, *args):
    prompt = f"""Formatiere die Pressemeldung exact so wie in den beispielen. beachte die formatierung der telefonnummern. Wichtig! liefere als ergebnis nur dei xml struktur so wie im beispiel. kommentiere nicht. füge keinerlei text am anfang oder ende hinzu. nur das xml zurückgeben. Danke!
Pressemeldung:\n{prompt}\n
Beispiele:
<Text><Sammler_Titel>SV Römershag hält<?ZE?>Jahresversammlung</Sammler_Titel><?ort?>Römershag<?_ort?>Der 1. SV Römershag 1949 hält seine Jahreshauptversammlung am Freitag, 9. Mai, um 19 Uhr im Sportheim Römershag. Auf der Tagesordnung stehen unter anderem die  Berichte der einzelnen Abteilungsleiter. Anträge sind bis spätestens Montag, 5. Mai, schriftlich beim 1. Vorsitzenden Thomas Dorn einzureichen.<?kü?>sek<?_kü?></Text>
<Text><Sammler_Titel>Seniorensprechstunde: Stadtrat bietet Hilfe</Sammler_Titel><?ort?>Bad Kissingen<?_ort?>Die Seniorenbeauftragte des Stadtrats Bad Kissingen, Martina Greubel, bietet wieder eine Seniorensprechstunde an. Diese findet jeden dritten Mittwoch im Monat von 15 bis 16 Uhr im Raum links neben dem Stadtsaal statt (barrierefreier Zugang vom Rathaushof). Die nächste Sprechstunde findet am Mittwoch, 19. März, statt. Für Fragen steht Martina Greubel unter Tel. 0171/895 53 39 zur Verfügung.<?kü?>sek<?_kü?></Text>
<Text><Sammler_Titel>Wirtshaussingen im<?ZE?>Karl-Straub-Haus</Sammler_Titel><?ort?>Geroda<?_ort?>Das Wirtshaussingen im Würzburger Karl-Straub-Haus findet am Donnerstag, 10. April, von 14 Uhr bis 19 Uhr statt. &#39;Franzl aus der Rhön&#39; begleitet die Veranstaltung mit der Gitarre. Zum Mitsingen von traditionellen und modernen Liedern werden Liedtexte in großer Schrift bereitgestellt. Der Eintritt ist frei. Barrierefreier Zugang und eine behindertengerechte Toilette sind vorhanden; zum Aussteigen kann kurzfristig an das Gasthaus (Am Farnsberg 116) herangefahren werden. Teilnehmende können zwischen 14 Uhr und 19 Uhr flexibel kommen und gehen. Angehörige werden gebeten, ältere oder eingeschränkte Personen zu unterstützen. Für Verpflegung sorgt das Team des Hauses. Rückfragen sind unter Tel. 01522/958 66 76 möglich.<?kü?>sek<?_kü?></Text>
"""
    #model = "gemini-2.5-pro-exp-03-25"
    model = "gemini-2.0-flash"
    #model = "gemini-2.0-pro-exp-02-05"
    #model = "gemini-2.0-flash-thinking-exp-01-21"
    contents = [
        types.Content(
            role="user",
            parts=[
                types.Part.from_text(text=f"{prompt}"),
            ],
        ),
    ]
    tools = [
        types.Tool(google_search=types.GoogleSearch())
    ]
    generate_content_config = types.GenerateContentConfig(
        temperature=0.35,
        top_p=0.95,
        top_k=40,
        #max_output_tokens=65536,
        max_output_tokens=8192,
        tools=tools,
        response_mime_type="text/plain",
    )

    response = ""
    for chunk in client.models.generate_content_stream(
        model=model,
        contents=contents,
        config=generate_content_config,
    ):
        response += chunk.text
        
    display(Markdown(response))
    return response


if __name__ == "__main__":
    iface = gr.ChatInterface(
        fn=generate,
        title=gr.Markdown("# Chatbot with Google Search"),
        description="Chatbot powered by Gemini-2.5 Pro and Google Search.",
    )
    iface.launch()