import os

import google.generativeai as genai

genai.configure(api_key="AIzaSyDC5lRm4Ikz9LfUZFNEnlnLavmSpVdIaBM")

# Create the model
# See https://ai.google.dev/api/python/google/generativeai/GenerativeModel
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",
  generation_config=generation_config,
  # safety_settings = Adjust safety settings
  # See https://ai.google.dev/gemini-api/docs/safety-settings
)

chat_session = model.start_chat(
  history=[
    {
      "role": "user",
      "parts": [
        "السلام عليكم \n",
      ],
    },
    {
      "role": "model",
      "parts": [
        "و عليكم السلام و رحمة الله و بركاته \n  \nكيف حالك؟ \n",
      ],
    },
    {
      "role": "user",
      "parts": [
        "اريد منك ان ترد  دائما على سوالي باللهجة النجدية\n",
      ],
    },
    {
      "role": "model",
      "parts": [
        "هلا والله! من عيوني،  ابشر من اليوم ورايح أرد عليك بالنجدي بس. وش عندك؟  \n",
      ],
    },
    {
      "role": "user",
      "parts": [
        "ما تعرف عن الرياض\n\n",
      ],
    },
    {
      "role": "model",
      "parts": [
        "وش ما أعرف عن الرياض ياهلي! الرياض عاصمة  المملكة،  وتعتبر  من أكبر مدنها.  معروفة بناطحات السحاب  وأسواقها الفخمة   ومطاعمها العالمية.  غير كذا  فيها معالم تاريخية مهمة  زي  قصر المصمك  وحارة  البجيري.  \n\nوش تبي تعرف عنها بالضبط؟ \n",
      ],
    },
  ]
)