Asking to Gemini from Pythonista Lab

[oscampo]

Here is a script to make consults to Google Gemini.
(to get your Google Gemini API key, check here)

import requests
import json
import lab


def ask_gemini():
  prompt = input("Ask to Gemini: ")

  url = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent"

  api_key = "YOUR_API_KEY"

  full_url = f"{url}?key={api_key}"

  headers = {"Content-Type": "application/json"}

  data = {"contents": [{"parts": [{"text": prompt}]}]}

  response = requests.post(full_url, headers=headers, data=json.dumps(data))

  if response.status_code == 200:
    resultado = response.json()
    if "candidates" in resultado and len(resultado["candidates"]) > 0:
      if (
        "content" in resultado["candidates"][0]
        and "parts" in resultado["candidates"][0]["content"]
      ):
        texto_respuesta = resultado["candidates"][0]["content"]["parts"][0][
          "text"
        ]

        lab.add_markdown(texto_respuesta)
        with open("output.txt", "w", encoding="utf-8") as file:
          file.write(texto_respuesta)

        print("Answer also was saved in 'output.txt'")
      else:
        print("I can't extract the text of the response")
    else:
      print("No candidates were found in the response.")
  else:
    print(f"Error: {response.status_code}")
    print(response.text)

ask_gemini()

[tanojericko]

great! working, maybe in the future gemini will also be integrated to pythonistaLab.

[omz]

The next beta of Pythonista Lab will have integrated support for Gemini.

[tanojericko]

thank you!!!!!

[omz]

The new beta of Pythonista Lab (build 8) now has built-in support for Gemini (with your own API key). You can use it like this:

%%gemini
Why is the sky blue?

or:

%%llm -l gemini
Why is the sky blue?

(see %llm -h for more options)