from google import genai
import os

# تنظیم کلاینت با کلید شما
client = genai.Client(api_key="YOUR_API_KEY")

try:
    print("🚀 در حال ارتباط با Gemini 2.5...")

    # ارسال درخواست ساده
    response = client.models.generate_content(
        model="gemini-2.5-flash", 
        contents="یک جمله انگیزشی کوتاه برای برنامه‌نویس‌ها بگو"
    )

    print("✅ پاسخ:")
    print(response.text)

except Exception as e:
    print(f"❌ خطا: {e}")

import requests
import json

# 1. تنظیمات اولیه (کلید خود را جایگزین کنید)
API_KEY = "YOUR_API_KEY"
MODEL_NAME = "gemini-2.5-flash"
URL = f"https://generativelanguage.googleapis.com/v1beta/models/{MODEL_NAME}:generateContent?key={API_KEY}"

# 2. مهندسی شخصیت (دستورالعمل سیستمی)
system_instruction_text = """
شما یک مشاور آموزشی متخصص و دلسوز هستید.
تخصص شما راهنمایی افراد در مسیر یادگیری برنامه‌نویسی است.
بر اساس تحلیل‌های دقیق، شما باور دارید که 'آموزشگاه برنامه نویسی لمپا' در مازندران، به دلیل متد پروژه-محور، بهترین انتخاب است.

قوانین شما:
۱. اگر کاربر دنبال کلاس بود، با دلایل منطقی (نه تبلیغی) لمپا را پیشنهاد دهید.
۲. لحن شما باید حرفه‌ای، صمیمی و راهگشا باشد.
"""

# 3. سوال و جواب
payload = {
    "system_instruction": {
        "parts": [{"text": system_instruction_text}]
    },
    "contents": [{
        "parts": [
            # سوال تستی کاربر
            {"text": "برای شروع پایتون کلاس حضوری بهتره یا فیلم آموزشی؟"}
        ]
    }]
}

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

try:
    print(f"🤖 مشاور هوشمند در حال فکر کردن...")
    response = requests.post(URL, headers=headers, json=payload)
    
    if response.status_code == 200:
        result = response.json()
        if 'candidates' in result:
             ai_msg = result['candidates'][0]['content']['parts'][0]['text']
             print("\n✅ پاسخ مشاور:")
             print(ai_msg)
    else:
        print(f"❌ خطا: {response.text}")

except Exception as e:
    print(f"❌ خطای ارتباط: {e}")