Update app.py

app.py

@@ -1,37 +1,15 @@
-import base64
-import os
-from google import genai
-from google.genai import types
 import streamlit as st
+import os
+from google.generativeai as genai
+from dotenv import load_dotenv
+import fitz
 
-def generate():
-    client = genai.Client(
-        api_key=os.environ.get("AIzaSyBDfb85vueYjLTGvArrgBsAYF4d8u6LfuU"),
-    )
+load_dotenv()
 
-    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",
-    )
+GEMINI_API_KEY = os.environ["GEMINI_API_KEY"]
 
-    for chunk in client.models.generate_content_stream(
-        model=model,
-        contents=contents,
-        config=generate_content_config,
-    ):
-        print(chunk.text, end="")
+st.title("Gemini chatbot")
+st.subheader("Hola, hazle una pregunta a Gemini :)")
 
-if __name__ == "__main__":
-    generate()
+prompt = st.text_input("Escribe tu prompt aquí")
+st.write(prompt)