Update app.py

app.py

@@ -1,32 +1,31 @@
-import base64
 import os
-from google import genai
-from google.genai import types
+import google.generativeai as genai
+from google.generativeai import types
 import gradio as gr
 
 def generate_text(user_input):
     try:
-        client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
+        # Initialize client with API key
+        genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
         
         model = "gemini-2.5-pro-exp-03-25"
         contents = [
             types.Content(
                 role="user",
-                parts=[
-                    types.Part.from_text(text=user_input),
-                ],
+                parts=[types.Part.from_text(user_input)],
             ),
         ]
-        generate_content_config = types.GenerateContentConfig(
+        config = types.GenerationConfig(
             temperature=2,
             response_mime_type="text/plain",
         )
 
+        # Generate response
         response = []
-        for chunk in client.models.generate_content_stream(
+        for chunk in genai.generate_content_stream(
             model=model,
             contents=contents,
-            config=generate_content_config,
+            generation_config=config,
         ):
             response.append(chunk.text)
             
@@ -35,13 +34,14 @@ def generate_text(user_input):
     except Exception as e:
         return f"Error: {str(e)}"
 
+# Create Gradio interface
 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"
+    inputs=gr.Textbox(lines=4, placeholder="Enter your prompt...", label="Input"),
+    outputs=gr.Textbox(label="Output"),
+    title="Gemini 2.5 Pro Demo",
+    description="Generate text with Gemini 2.5 Pro (streaming enabled)",
+    allow_flagging="never",
 )
 
 if __name__ == "__main__":