Update app.py

app.py

@@ -1,47 +1,40 @@
 import os
-import time
 import gradio as gr
 from google import genai
+from google.genai import types
 
-# Initialize client with API key from env var
-client = genai.Client()
+# Load API key from environment variable
+API_KEY = os.getenv("API_KEY")
+if not API_KEY:
+    raise ValueError("API_KEY environment variable is not set")
 
-def generate_video(prompt):
-    operation = client.models.generate_videos(
-        model="veo-3.0-generate-preview",
-        prompt=prompt,
+# Initialize the genai client with API key
+client = genai.Client(api_key=API_KEY)
+
+def generate_grounded_response(prompt):
+    grounding_tool = types.Tool(
+        google_search=types.GoogleSearch()
+    )
+
+    config = types.GenerateContentConfig(
+        tools=[grounding_tool]
     )
 
-    while not operation.done:
-        print("Waiting for video generation to complete...")
-        time.sleep(10)
-        operation = client.operations.get(operation)
-
-    generated_video = operation.response.generated_videos[0]
-    # Download video bytes
-    video_file = client.files.download(file=generated_video.video)
-    
-    # Save locally
-    video_path = "dialogue_example.mp4"
-    with open(video_path, "wb") as f:
-        f.write(video_file)
-    
-    return video_path
+    response = client.models.generate_content(
+        model="gemini-2.5-flash",
+        contents=prompt,
+        config=config,
+    )
+
+    return response.text
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Gemini Veo Video Generator")
-    prompt_input = gr.Textbox(label="Video Prompt", value="""A close up of two people staring at a cryptic drawing on a wall, torchlight flickering.
-A man murmurs, 'This must be it. That's the secret code.' The woman looks at him and whispering excitedly, 'What did you find?'""")
-    generate_btn = gr.Button("Generate Video")
-    video_output = gr.Video(label="Generated Video")
-    status = gr.Textbox(label="Status")
-
-    def on_generate(prompt):
-        status.update("Generating video, please wait...")
-        video_path = generate_video(prompt)
-        status.update("Generation complete!")
-        return video_path, "Generation complete!"
-
-    generate_btn.click(on_generate, inputs=prompt_input, outputs=[video_output, status])
+    gr.Markdown("# Gemini 2.5 Flash Grounded Content Generator")
+
+    prompt_input = gr.Textbox(label="Enter your question or prompt", lines=3, value="Who won the euro 2024?")
+    generate_btn = gr.Button("Generate Response")
+    output_text = gr.Textbox(label="Grounded Response", lines=10)
+
+    generate_btn.click(fn=generate_grounded_response, inputs=prompt_input, outputs=output_text)
 
 demo.launch()