Update app.py

app.py

@@ -4,68 +4,13 @@ import os
 import json
 from google import genai
 from google.genai import types
-from gradio_client import Client
-
-def clean_json_string(json_str):
-    """
-    Removes any comments or prefixes before the actual JSON content.
-    """
-    # Find the first occurrence of '{'
-    json_start = json_str.find('{')
-    if json_start == -1:
-        # If no '{' is found, try with '[' for arrays
-        json_start = json_str.find('[')
-        if json_start == -1:
-            return json_str
-    
-    # Extract everything from the first JSON marker
-    cleaned_str = json_str[json_start:]
-    
-    # Verify it's valid JSON
-    try:
-        json.loads(cleaned_str)
-        return cleaned_str
-    except json.JSONDecodeError:
-        return cleaned_str # Return extracted part anyway
-
-def generate(input_text):
-    client = genai.Client(
-        api_key=os.environ.get("GEMINI_API_KEY"),
-    )
-
-    model = "gemini-flash-latest"
-    contents = [
-        types.Content(
-            role="user",
-            parts=[
-                types.Part.from_text(text=f"{input_text}"),
-            ],
-        ),
-    ]
-    tools = [
-        types.Tool(googleSearch=types.GoogleSearch(
-        )),
-    ]
-    generate_content_config = types.GenerateContentConfig(
-        temperature=0.35,
-        tools=tools,
-    )
-
-    for chunk in client.models.generate_content_stream(
-        model=model,
-        contents=contents,
-        config=generate_content_config,
-    ):
-        print(chunk.text, end="")
-
-
 
 def generate1(input_text):
     client = genai.Client(
         api_key=os.environ.get("GEMINI_API_KEY"),
     )
 
-    model = "gemini-2.0-flash"
+    model = "gemini-flash-latest"
     contents = [
         types.Content(
             role="user",
@@ -74,9 +19,12 @@ def generate1(input_text):
             ],
         ),
     ]
+    
+    # Updated tool config for Gemini 2.0
     tools = [
-        types.Tool(google_search_retrieval=types.DynamicRetrievalConfig(dynamic_threshold=0.3, mode="MODE_DYNAMIC")),
+        types.Tool(google_search_retrieval=types.DynamicRetrievalConfig(mode="DYNAMIC"))
     ]
+    
     generate_content_config = types.GenerateContentConfig(
         temperature=0.45,
         tools=tools,
@@ -84,26 +32,29 @@ def generate1(input_text):
     )
 
     full_response = ""
+    # We yield two values: the accumulated response and an empty string (to clear the input)
     for chunk in client.models.generate_content_stream(
         model=model,
         contents=contents,
         config=generate_content_config,
     ):
-        full_response += chunk.text
-        yield full_response
+        if chunk.text:
+            full_response += chunk.text
+        # Yielding [Output, Input]
+        yield full_response, gr.update(value="") 
 
 if __name__ == '__main__':
     with gr.Blocks() as demo:
         gr.Markdown("# Gemini 2.0 Flash + Websearch")
         output_textbox = gr.Markdown()
-        input_textbox = gr.Textbox(lines=3, label="Input", placeholder="Enter event details here...")
+        input_textbox = gr.Textbox(lines=3, label="Input", placeholder="Enter name here...")
         submit_button = gr.Button("Send")
         
-        # Die Funktion generate gibt nun (Response, ClearInput) zurück
+        # Pointing to generate1 which now yields two values
         submit_button.click(
-            fn=generate,
+            fn=generate1,
             inputs=input_textbox,
-            outputs=[output_textbox]
+            outputs=[output_textbox, input_textbox]
         )
         
     demo.launch(show_error=True)