Update app.py

app.py

@@ -16,53 +16,56 @@ def clean_json_string(json_str):
         # If no '{' is found, try with '[' for arrays
         json_start = json_str.find('[')
         if json_start == -1:
-            return json_str  # Return original if no JSON markers found
+            return json_str
     
     # Extract everything from the first JSON marker
     cleaned_str = json_str[json_start:]
-    return cleaned_str
+    
     # Verify it's valid JSON
     try:
         json.loads(cleaned_str)
         return cleaned_str
     except json.JSONDecodeError:
-        return json_str  # Return original if cleaning results in invalid JSON
+        return cleaned_str # Return extracted part anyway
 
 def generate(input_text):
     try:
         client = genai.Client(
             api_key=os.environ.get("GEMINI_API_KEY"),
         )
-    except Exception as e:
-        return f"Error initializing client: {e}.  Make sure GEMINI_API_KEY is set."
-
-    model = "gemini-3-flash-preview"
+        
+        model = "gemini-2.0-flash" # Hinweis: gemini-3 existiert aktuell noch nicht
         contents = [
             types.Content(
                 role="user",
                 parts=[
-                    types.Part.from_text(text="""INSERT_INPUT_HERE"""),
+                    types.Part.from_text(text=input_text),
                 ],
             ),
         ]
         tools = [
-            types.Tool(googleSearch=types.GoogleSearch(
-            )),
+            types.Tool(google_search=types.GoogleSearch()),
         ]
         generate_content_config = types.GenerateContentConfig(
             temperature=0.35,
-            thinking_config=types.ThinkingConfig(
-                thinking_level="HIGH",
-            ),
+            # thinking_config ist aktuell nur für spezielle Modelle verfügbar
+            # thinking_config=types.ThinkingConfig(thinking_level="HIGH"), 
             tools=tools,
         )
 
+        full_response = ""
         for chunk in client.models.generate_content_stream(
             model=model,
             contents=contents,
             config=generate_content_config,
         ):
-            print(chunk.text, end="")
+            if chunk.text:
+                full_response += chunk.text
+                # Wir geben zwei Werte zurück, da Gradio zwei Outputs erwartet
+                yield full_response, "" 
+
+    except Exception as e:
+        yield f"Error: {e}", input_text
 
 
 def generate1(input_text):
@@ -70,17 +73,17 @@ def generate1(input_text):
         api_key=os.environ.get("GEMINI_API_KEY"),
     )
 
-    model = "gemini-3-flash-preview"
+    model = "gemini-2.0-flash"
     contents = [
         types.Content(
             role="user",
             parts=[
-                types.Part.from_text(text=f"return json object with keys name and email. name = {input_text}  search the web for the email value. return json object only, no additional text or comments"),
+                types.Part.from_text(text=f"return json object with keys name and email. name = {input_text} search the web for the email value. return json object only, no additional text or comments"),
             ],
         ),
     ]
     tools = [
-        types.Tool(googleSearchRetrieval=types.DynamicRetrievalConfig(dynamicThreshold=0.3, mode=types.DynamicRetrievalConfigMode.MODE_DYNAMIC)),
+        types.Tool(google_search_retrieval=types.DynamicRetrievalConfig(dynamic_threshold=0.3, mode="MODE_DYNAMIC")),
     ]
     generate_content_config = types.GenerateContentConfig(
         temperature=0.45,
@@ -88,20 +91,27 @@ def generate1(input_text):
         response_mime_type="text/plain",
     )
 
+    full_response = ""
     for chunk in client.models.generate_content_stream(
         model=model,
         contents=contents,
         config=generate_content_config,
     ):
-        print(chunk.text, end="")
-    
+        full_response += chunk.text
+        yield full_response
 
 if __name__ == '__main__':
-
     with gr.Blocks() as demo:
-        title=gr.Markdown("# Gemini 2.0 Flash + Websearch")
+        gr.Markdown("# Gemini 2.0 Flash + Websearch")
         output_textbox = gr.Markdown()
-        input_textbox = gr.Textbox(lines=3, label="", placeholder="Enter event details here...")
-        submit_button = gr.Button("send")
-        submit_button.click(fn=generate,inputs=input_textbox,outputs=[output_textbox, input_textbox])
+        input_textbox = gr.Textbox(lines=3, label="Input", placeholder="Enter event details here...")
+        submit_button = gr.Button("Send")
+        
+        # Die Funktion generate gibt nun (Response, ClearInput) zurück
+        submit_button.click(
+            fn=generate,
+            inputs=input_textbox,
+            outputs=[output_textbox, input_textbox]
+        )
+        
     demo.launch(show_error=True)