Update app.py

app.py

@@ -4,190 +4,252 @@ from datetime import datetime
 from passlib.hash import bcrypt
 import os
 
+# تهيئة الصفحة
 st.set_page_config(page_title="نظام إدارة الشكاوى", layout="centered")
 
-# الاتصال بقاعدة البيانات
-conn = sqlite3.connect("complaints_system.db", check_same_thread=False)
-c = conn.cursor()
-
-# إنشاء الجداول
-c.execute('''CREATE TABLE IF NOT EXISTS users (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    username TEXT UNIQUE,
-    password TEXT,
-    role TEXT
-)''')
-
-c.execute('''CREATE TABLE IF NOT EXISTS complaints (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    complaint_id TEXT UNIQUE,
-    username TEXT,
-    complaint_text TEXT,
-    status TEXT,
-    response TEXT,
-    timestamp TEXT,
-    user_comment TEXT
-)''')
-
-conn.commit()
-
-# تهيئة الجلسة
-if "is_logged_in" not in st.session_state:
-    st.session_state.is_logged_in = False
-if "user_type" not in st.session_state:
-    st.session_state.user_type = None
-if "username" not in st.session_state:
-    st.session_state.username = ""
-if "complaint_input" not in st.session_state:
-    st.session_state.complaint_input = ""
-
-# تسجيل الدخول
-if not st.session_state.is_logged_in:
+# إدارة اتصال قاعدة البيانات
+def get_db_connection():
+    conn = sqlite3.connect("complaints_system.db", check_same_thread=False)
+    conn.row_factory = sqlite3.Row
+    return conn
+
+# إنشاء الجداول إذا لم تكن موجودة
+def init_db():
+    conn = get_db_connection()
+    conn.execute('''CREATE TABLE IF NOT EXISTS users (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        username TEXT UNIQUE,
+        password TEXT,
+        role TEXT
+    )''')
+    
+    conn.execute('''CREATE TABLE IF NOT EXISTS complaints (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        complaint_id TEXT UNIQUE,
+        username TEXT,
+        complaint_text TEXT,
+        status TEXT,
+        response TEXT,
+        timestamp TEXT,
+        user_comment TEXT
+    )''')
+    conn.commit()
+    conn.close()
+
+init_db()
+
+# إدارة حالة الجلسة
+def init_session():
+    if "auth" not in st.session_state:
+        st.session_state.auth = {
+            "is_logged_in": False,
+            "username": "",
+            "role": ""
+        }
+
+init_session()
+
+# واجهة تسجيل الدخول
+def login_page():
     st.title("🔐 تسجيل الدخول")
     username = st.text_input("اسم المستخدم")
     password = st.text_input("كلمة المرور", type="password")
 
     if st.button("تسجيل الدخول"):
-        c.execute("SELECT * FROM users WHERE username=?", (username,))
-        user = c.fetchone()
-        if user:
-            if bcrypt.verify(password, user[2]):
-                st.session_state.is_logged_in = True
-                st.session_state.username = username
-                st.session_state.user_type = "sender" if user[3] == "sender" else "receiver"
-                st.rerun()
-            else:
-                st.error("❌ كلمة المرور غير صحيحة.")
+        conn = get_db_connection()
+        user = conn.execute("SELECT * FROM users WHERE username = ?", (username,)).fetchone()
+        conn.close()
+        
+        if user and bcrypt.verify(password, user["password"]):
+            st.session_state.auth = {
+                "is_logged_in": True,
+                "username": user["username"],
+                "role": user["role"]
+            }
+            st.rerun()
+        elif user:
+            st.error("❌ كلمة المرور غير صحيحة")
         else:
-            st.warning("❌ اسم المستخدم غير موجود.")
+            st.warning("❌ اسم المستخدم غير موجود")
 
-# مقدم الشكوى
-elif st.session_state.user_type == "sender":
+# صفحة مقدم الشكوى
+def sender_page():
     st.title("📨 إرسال شكوى جديدة")
+    complaint_text = st.text_area("📝 وصف الشكوى", key="complaint_text")
     
-    def clear_complaint():
-        st.session_state.complaint_input = ""
-    
-    complaint_text = st.text_area("📝 وصف الشكوى", key="complaint_input")
-    
-    if st.button("📤 إرسال الشكوى", on_click=clear_complaint):
-        if not complaint_text or not complaint_text.strip():
-            st.warning("⚠️ الرجاء إدخال نص الشكوى.")
+    if st.button("📤 إرسال الشكوى"):
+        if not complaint_text.strip():
+            st.warning("⚠️ الرجاء إدخال نص الشكوى")
         else:
-            c.execute("SELECT * FROM complaints WHERE username=?", (st.session_state.username,))
-            user_complaints = c.fetchall()
-            prefix = st.session_state.username[:4].lower()
-            next_id = len(user_complaints) + 1
-            complaint_id = f"{prefix}_{next_id:07}"
-
-            c.execute("SELECT * FROM complaints WHERE complaint_id=?", (complaint_id,))
-            if c.fetchone():
-                st.warning("⚠️ تم إرسال هذه الشكوى مسبقًا بناءً على المعرف!")
-            else:
-                timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-                c.execute("""INSERT INTO complaints 
-                          (complaint_id, username, complaint_text, status, response, timestamp, user_comment) 
-                          VALUES (?, ?, ?, ?, ?, ?, ?)""",
-                          (complaint_id, st.session_state.username, complaint_text.strip(), 
-                           "Pending", "", timestamp, ""))
+            conn = get_db_connection()
+            try:
+                # إنشاء معرف فريد للشكوى
+                user_complaints = conn.execute(
+                    "SELECT COUNT(*) FROM complaints WHERE username = ?",
+                    (st.session_state.auth["username"],)
+                ).fetchone()[0]
+                
+                complaint_id = f"{st.session_state.auth['username'][:4]}_{user_complaints + 1:07d}"
+                
+                conn.execute(
+                    """INSERT INTO complaints 
+                    (complaint_id, username, complaint_text, status, response, timestamp, user_comment) 
+                    VALUES (?, ?, ?, ?, ?, ?, ?)""",
+                    (complaint_id, st.session_state.auth["username"], complaint_text.strip(),
+                     "Pending", "", datetime.now().strftime("%Y-%m-%d %H:%M:%S"), "")
+                )
                 conn.commit()
                 st.success("✅ تم تقديم الشكوى بنجاح")
+                st.session_state.complaint_text = ""  # مسح حقل الإدخال
                 st.rerun()
+            except sqlite3.IntegrityError:
+                st.error("❌ حدث خطأ أثناء حفظ الشكوى")
+            finally:
+                conn.close()
 
+    # عرض الشكاوى السابقة
     st.subheader("📋 الشكاوى السابقة")
-    c.execute("SELECT * FROM complaints WHERE username=? ORDER BY id DESC", (st.session_state.username,))
-    complaints = c.fetchall()
-
-    for row in complaints:
-        comp_id, username, text, status, response, timestamp, user_comment = row[1:]
-        with st.expander(f"📨 {comp_id} | الحالة: {status}"):
-            st.write(f"📅 تم الإرسال في: {timestamp}")
-            st.write(f"📄 الشكوى: {text}")
-
-            if status.lower() == "pending":
-                new_text = st.text_area("📝 تعديل الشكوى", value=text, key=f"edit_{comp_id}")
-                if st.button("💾 حفظ التعديل", key=f"save_{comp_id}"):
+    conn = get_db_connection()
+    complaints = conn.execute(
+        "SELECT * FROM complaints WHERE username = ? ORDER BY id DESC",
+        (st.session_state.auth["username"],)
+    ).fetchall()
+    conn.close()
+
+    for complaint in complaints:
+        with st.expander(f"📨 {complaint['complaint_id']} | الحالة: {complaint['status']}"):
+            st.write(f"📅 تم الإرسال في: {complaint['timestamp']}")
+            st.write(f"📄 الشكوى: {complaint['complaint_text']}")
+
+            if complaint["status"].lower() == "pending":
+                new_text = st.text_area(
+                    "📝 تعديل الشكوى",
+                    value=complaint["complaint_text"],
+                    key=f"edit_{complaint['complaint_id']}"
+                )
+                if st.button("💾 حفظ التعديل", key=f"save_{complaint['complaint_id']}"):
                     if new_text.strip():
-                        c.execute("UPDATE complaints SET complaint_text=? WHERE complaint_id=?", 
-                                (new_text.strip(), comp_id))
+                        conn = get_db_connection()
+                        conn.execute(
+                            "UPDATE complaints SET complaint_text = ? WHERE complaint_id = ?",
+                            (new_text.strip(), complaint["complaint_id"])
+                        )
                         conn.commit()
+                        conn.close()
                         st.success("✅ تم تعديل الشكوى")
                         st.rerun()
                     else:
                         st.warning("⚠️ لا يمكن حفظ شكوى فارغة")
             else:
-                if response:
-                    st.success(f"💬 الرد: {response}")
-                    if not user_comment:
-                        comment = st.text_area("✏️ تعليقك على الرد", key=f"comment_{comp_id}")
-                        if st.button("📩 إرسال التعليق", key=f"send_comment_{comp_id}"):
+                if complaint["response"]:
+                    st.success(f"💬 الرد: {complaint['response']}")
+                    
+                    if not complaint["user_comment"]:
+                        comment = st.text_area(
+                            "✏️ تعليقك على الرد",
+                            key=f"comment_{complaint['complaint_id']}"
+                        )
+                        if st.button("📩 إرسال التعليق", key=f"send_comment_{complaint['complaint_id']}"):
                             if comment.strip():
-                                c.execute("UPDATE complaints SET user_comment=? WHERE complaint_id=?", 
-                                        (comment.strip(), comp_id))
+                                conn = get_db_connection()
+                                conn.execute(
+                                    "UPDATE complaints SET user_comment = ? WHERE complaint_id = ?",
+                                    (comment.strip(), complaint["complaint_id"])
+                                )
                                 conn.commit()
+                                conn.close()
                                 st.success("✅ تم إرسال التعليق بنجاح")
                                 st.rerun()
                             else:
                                 st.warning("⚠️ لا يمكن إرسال تعليق فارغ")
                     else:
-                        st.info(f"🗣️ تعليقك السابق: {user_comment}")
+                        st.info(f"🗣️ تعليقك السابق: {complaint['user_comment']}")
 
-# مستقبل الشكوى (الـ admin أو الموظف)
-elif st.session_state.user_type == "receiver":
+# صفحة مستقبل الشكاوى
+def receiver_page():
     st.title("📬 إدارة الشكاوى")
-
-    c.execute("SELECT COUNT(*) FROM complaints WHERE status = 'Pending'")
-    pending_count = c.fetchone()[0]
+    
+    conn = get_db_connection()
+    pending_count = conn.execute(
+        "SELECT COUNT(*) FROM complaints WHERE status = 'Pending'"
+    ).fetchone()[0]
+    
     if pending_count > 0:
         st.warning(f"🚨 يوجد {pending_count} شكوى جديدة لم يتم التعامل معها!")
-
-    c.execute("SELECT * FROM complaints ORDER BY id DESC")
-    complaints = c.fetchall()
+    
+    complaints = conn.execute(
+        "SELECT * FROM complaints ORDER BY id DESC"
+    ).fetchall()
+    conn.close()
 
     if not complaints:
         st.info("لا توجد شكاوى حالياً.")
     else:
         st.subheader("📋 جميع الشكاوى")
-        options = {f"{row[1]} - {row[2]}": row for row in complaints}
-        selected = st.selectbox("اختر شكوى للتعامل معها", list(options.keys()))
-        selected_row = options[selected]
-
-        st.subheader(f"📨 معرف الشكوى: {selected_row[1]}")
-        st.write(f"👤 مقدم الشكوى: {selected_row[2]}")
-        st.write(f"📝 الشكوى: {selected_row[3]}")
-        st.write(f"📅 الوقت: {selected_row[6]}")
-        if selected_row[7]:
-            st.info(f"🗣️ تعليق المستخدم على الرد: {selected_row[7]}")
-
-        new_status = st.selectbox("🔄 تحديث الحالة", 
-                                ["Pending", "In Progress", "Resolved"], 
-                                index=["Pending", "In Progress", "Resolved"].index(selected_row[4]))
-        new_response = st.text_area("💬 الرد على الشكوى", value=selected_row[5] or "")
+        selected_complaint = st.selectbox(
+            "اختر شكوى للتعامل معها",
+            complaints,
+            format_func=lambda x: f"{x['complaint_id']} - {x['username']}"
+        )
+
+        st.subheader(f"📨 معرف الشكوى: {selected_complaint['complaint_id']}")
+        st.write(f"👤 مقدم الشكوى: {selected_complaint['username']}")
+        st.write(f"📝 الشكوى: {selected_complaint['complaint_text']}")
+        st.write(f"📅 الوقت: {selected_complaint['timestamp']}")
+        
+        if selected_complaint["user_comment"]:
+            st.info(f"🗣️ تعليق المستخدم على الرد: {selected_complaint['user_comment']}")
+
+        new_status = st.selectbox(
+            "🔄 تحديث الحالة",
+            ["Pending", "In Progress", "Resolved"],
+            index=["Pending", "In Progress", "Resolved"].index(selected_complaint["status"])
+        )
+        
+        new_response = st.text_area(
+            "💬 الرد على الشكوى",
+            value=selected_complaint["response"] or ""
+        )
 
         if st.button("💾 حفظ التعديلات"):
-            c.execute("UPDATE complaints SET status=?, response=? WHERE complaint_id=?", 
-                     (new_status, new_response.strip(), selected_row[1]))
+            conn = get_db_connection()
+            conn.execute(
+                "UPDATE complaints SET status = ?, response = ? WHERE complaint_id = ?",
+                (new_status, new_response.strip(), selected_complaint["complaint_id"])
+            )
             conn.commit()
+            conn.close()
             st.success("✅ تم حفظ التعديلات")
             st.rerun()
 
-# زر تسجيل الخروج دائم
-if st.session_state.get("is_logged_in"):
-    st.markdown("---")
-    if st.button("🚪 تسجيل ا��خروج"):
-        st.session_state.is_logged_in = False
-        st.session_state.user_type = None
-        st.session_state.username = ""
-        st.rerun()
-
-# زر لتحميل قاعدة البيانات
-if os.path.exists("complaints_system.db"):
-    with open("complaints_system.db", "rb") as db_file:
-        st.download_button(
-            label="📥 تحميل قاعدة البيانات",
-            data=db_file,
-            file_name="complaints_system.db",
-            mime="application/octet-stream"
-        )
-else:
-    st.warning("⚠️ ملف قاعدة البيانات غير موجود.")
+# الواجهة الرئيسية
+def main():
+    if not st.session_state.auth["is_logged_in"]:
+        login_page()
+    else:
+        if st.session_state.auth["role"] == "sender":
+            sender_page()
+        else:
+            receiver_page()
+
+        st.markdown("---")
+        if st.button("🚪 تسجيل الخروج"):
+            st.session_state.auth = {
+                "is_logged_in": False,
+                "username": "",
+                "role": ""
+            }
+            st.rerun()
+
+    # زر تحميل قاعدة البيانات
+    if os.path.exists("complaints_system.db"):
+        with open("complaints_system.db", "rb") as db_file:
+            st.download_button(
+                label="📥 تحميل قاعدة البيانات",
+                data=db_file,
+                file_name="complaints_system.db",
+                mime="application/octet-stream"
+            )
+
+if __name__ == "__main__":
+    main()