import os
import google.generativeai as genai

genai.configure(api_key="AIzaSyD1TJ82gJaI9fL239kdsk6d4t-oLL3G2PE")  # Remplacez par votre clé API

model = genai.GenerativeModel(model_name='gemini-1.5-pro', tools=[genai.Tool.CODE_EXECUTION])

chat = model.start_chat()

# Première requête: Demander le code
response = chat.send_message("What is the sum of the first 50 prime numbers? Generate Python code for the calculation, and make sure you get all 50.")

# Extraire le code de la réponse (Gemini devrait le fournir dans un bloc de code)
code_block = None
for tool_code in response.tool_code:
    if tool_code.type == 'code':
        code_block = tool_code
        break

if code_block:
    # Exécuter le code avec l'outil code_execution
    execution_response = chat.send_message(code_block.code)

    # Afficher le résultat de l'exécution
    print(execution_response.text)
else:
    print("Gemini n'a pas retourné de code exécutable.")