import pandas as pd
from huggingface_hub import HfApi, hf_hub_download
import os
from datetime import datetime

# إعدادات الاتصال
DATASET_ID = "اسم_حسابك/epssar-database" # غير هذا لاسم حسابك
FILE_NAME = "security_logs.csv"
TOKEN = os.getenv("HF_TOKEN")

def save_to_database(tweet_text, analysis_result, risk_level):
    """حفظ نتائج التحليل في قاعدة البيانات على هقنق فيس"""
    try:
        # 1. محاولة تحميل الملف الحالي إذا كان موجوداً
        try:
            file_path = hf_hub_download(repo_id=DATASET_ID, filename=FILE_NAME, repo_type="dataset", token=TOKEN)
            df = pd.read_csv(file_path)
        except:
            # إذا كان أول مرة، ننشئ جدول جديد
            df = pd.DataFrame(columns=["date", "tweet", "analysis", "risk_level"])

        # 2. إضافة البيانات الجديدة
        new_entry = {
            "date": datetime.now().strftime("%Y-%m-%d %H:%M"),
            "tweet": tweet_text,
            "analysis": analysis_result,
            "risk_level": risk_level
        }
        df = pd.concat([df, pd.DataFrame([new_entry])], ignore_index=True)

        # 3. حفظ الملف محلياً ثم رفعه للهقنق فيس
        df.to_csv(FILE_NAME, index=False)
        api = HfApi()
        api.upload_file(
            path_or_fileobj=FILE_NAME,
            path_in_repo=FILE_NAME,
            repo_id=DATASET_ID,
            repo_type="dataset",
            token=TOKEN
        )
        return True
    except Exception as e:
        print(f"خطأ في قاعدة البيانات: {e}")
        return False