Update app.py

app.py

@@ -3,13 +3,52 @@ import os
 from google.generativeai as genai
 from dotenv import load_dotenv
 import fitz
+import base64
 
 load_dotenv()
 
 GEMINI_API_KEY = os.environ["GEMINI_API_KEY"]
 
+
+def generate():
+    client = genai.Client(
+        api_key=os.environ.get("GEMINI_API_KEY"),
+    )
+
+    model = "gemini-2.5-pro-exp-03-25"
+    contents = [
+        types.Content(
+            role="user",
+            parts=[
+                types.Part.from_text(text="""INSERT_INPUT_HERE"""),
+            ],
+        ),
+    ]
+    generate_content_config = types.GenerateContentConfig(
+        temperature=1,
+        top_p=0.95,
+        top_k=64,
+        max_output_tokens=65536,
+        response_mime_type="text/plain",
+    )
+
+    for chunk in client.models.generate_content_stream(
+        model=model,
+        contents=contents,
+        config=generate_content_config,
+    ):
+        print(chunk.text, end="")
+
+if __name__ == "__main__":
+    generate()
+
+
+
 st.title("Gemini chatbot")
 st.subheader("Hola, hazle una pregunta a Gemini :)")
 
 prompt = st.text_input("Escribe tu prompt aquí")
-st.write(prompt)
+
+if prompt:
+    response = model.generate(prompt)
+    st.write(response.text)