main.py

import os
import google.generativeai as genai
from dotenv import load_dotenv
from google.generativeai.types import HarmCategory, HarmBlockThreshold

load_dotenv()  # Load environment variables from .env
genai.configure(api_key=os.environ["GEMINI_API_KEY"])

# Create the model
generation_config = {
  "temperature": 1,
  "top_p": 0.95,
  "top_k": 64,
  "max_output_tokens": 8192,
  "response_mime_type": "text/plain",
}

model = genai.GenerativeModel(
  model_name="gemini-1.5-pro-exp-0801",
  generation_config=generation_config,
  # Adjust safety settings
  # See https://ai.google.dev/gemini-api/docs/safety-settings
  safety_settings = {
        HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
        HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
        HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
        HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
  }
)

chat_session = model.start_chat(
  history=[
  ]
)

response = chat_session.send_message("Give me a high risk stock portfolio out of ETFs available in the UK as diversified as possible")

print(response.text)

# Write the response text to the file
output_filename = "gemini_response.md"
with open(output_filename, "a", encoding="utf-8") as output_file:
    output_file.write(response.text)
print(f"Response appended to: {output_filename}")