File size: 2,179 Bytes
ce8ec42 67495e4 ce8ec42 7f90415 ce8ec42 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
import streamlit as st
from google import genai
from google.genai.types import GenerateContentConfig, ThinkingConfig
def main():
st.title("Interface Gemini API avec Thoughts")
# Zone de saisie pour la clé API
api_key = st.text_input("Entrez votre clé API Google", type="password")
# Zone de texte pour le prompt
prompt = st.text_area("Entrez votre prompt")
if st.button("Générer"):
if not api_key:
st.error("Veuillez entrer une clé API")
return
if not prompt:
st.warning("Veuillez entrer un prompt")
return
try:
# Configuration du client
client = genai.Client(
api_key=api_key,
http_options={'api_version':'v1alpha'}
)
# Configuration pour les thoughts
config = GenerateContentConfig(
thinking_config=ThinkingConfig(include_thoughts=True)
)
# Affichage d'un spinner pendant la génération
with st.spinner("Génération en cours..."):
response = client.models.generate_content(
model="gemini-2.0-flash-thinking-exp-01-21",
contents=prompt,
config=config,
)
# Affichage des résultats dans des sections distinctes
st.subheader("Résultats")
print(response)
for part in response.candidates[0].content.parts:
if part.thought:
with st.expander("💭 Thoughts", expanded=True):
st.write(part.text)
else:
with st.expander("📝 Réponse", expanded=True):
st.write(part.text)
except Exception as e:
st.error(f"Une erreur s'est produite: {str(e)}")
if __name__ == "__main__":
# Configuration de la page
st.set_page_config(
page_title="Interface Gemini API",
page_icon="🤖",
layout="wide"
)
main() |