Update database_manager.py

database_manager.py

@@ -1,44 +1,51 @@
-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}")
+import pandas as pd
+from huggingface_hub import HfApi, hf_hub_download
+import os
+from datetime import datetime
+
+# إعدادات الاتصال (تأكد من إنشاء Dataset بهذا الاسم في حسابك أولاً)
+DATASET_ID = "MZ14E/epssar-database" 
+FILE_NAME = "security_logs.csv"
+LOCAL_PATH = f"/tmp/{FILE_NAME}" # الحفظ في المجلد المؤقت للسيرفر
+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 Exception:
+            # إذا كان أول مرة أو الملف غير موجود، ننشئ جدول جديد
+            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(LOCAL_PATH, index=False)
+        
+        api = HfApi()
+        api.upload_file(
+            path_or_fileobj=LOCAL_PATH,
+            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