Update app.py

app.py

@@ -1,93 +1,112 @@
 import streamlit as st
 import pandas as pd
+import os
+import uuid
+from datetime import datetime
 
-# إعدادات الملف
-file_path = "complaints.csv"
+st.set_page_config(page_title="نظام الشكاوى", layout="centered")
 
-# تحميل الشكاوى من ملف CSV أو إنشاء الملف إذا لم يكن موجود
-try:
-    df = pd.read_csv(file_path)
-except FileNotFoundError:
-    df = pd.DataFrame(columns=["مقدم الشكوى", "وصف الشكوى", "حالة الشكوى", "رد الشكوى"])
+# اسم الملف لتخزين الشكاوى
+csv_file = "complaints.csv"
 
-# صفحة تسجيل الدخول
-def login_screen():
-    st.subheader("تسجيل الدخول")
-    user_type = st.selectbox("اختر نوع المستخدم:", ["مقدم شكوى", "مستقبل شكوى"])
-    username = st.text_input("اسم المستخدم:")
-    password = st.text_input("كلمة المرور:", type="password")
+# التأكد من وجود الملف
+if not os.path.exists(csv_file):
+    df_init = pd.DataFrame(columns=[
+        "complaint_id", "username", "complaint_text", "status", "response", "timestamp"
+    ])
+    df_init.to_csv(csv_file, index=False)
+
+def load_data():
+    return pd.read_csv(csv_file)
+
+def save_data(df):
+    df.to_csv(csv_file, index=False)
+
+def generate_complaint_id(user_id, existing_ids):
+    counter = 1
+    while True:
+        cid = f"SUE U{user_id}-{str(counter).zfill(6)}"
+        if cid not in existing_ids:
+            return cid
+        counter += 1
+
+# جلسة المستخدم
+if "logged_in" not in st.session_state:
+    st.session_state.logged_in = False
+if "role" not in st.session_state:
+    st.session_state.role = None
+if "username" not in st.session_state:
+    st.session_state.username = ""
+
+# تسجيل الدخول
+if not st.session_state.logged_in:
+    st.title("تسجيل الدخول")
+    username = st.text_input("اسم المستخدم")
+    password = st.text_input("كلمة المرور", type="password")
+    role = st.selectbox("الدور", ["مقدم الشكوى", "مستقبل الشكوى"])
+    
     if st.button("دخول"):
         if username and password:
-            # حفظ حالة المستخدم في session_state
-            st.session_state.user_type = user_type
+            st.session_state.logged_in = True
+            st.session_state.role = role
             st.session_state.username = username
-            return True
+            st.rerun()
         else:
-            st.error("الرجاء إدخال اسم المستخدم وكلمة المرور.")
-    return False
+            st.error("من فضلك أدخل اسم المستخدم وكلمة المرور")
 
-# التحقق من حالة تسجيل الدخول
-if "user_type" not in st.session_state:
-    if login_screen():
-        st.rerun()  # إعادة تحميل الصفحة بعد تسجيل الدخول
 else:
-    user_type = st.session_state.user_type
-    username = st.session_state.username
+    st.sidebar.success(f"مرحبًا {st.session_state.username} ({st.session_state.role})")
+    if st.sidebar.button("تسجيل الخروج"):
+        st.session_state.logged_in = False
+        st.session_state.role = None
+        st.session_state.username = ""
+        st.rerun()
 
-    if user_type == "مقدم شكوى":
-        st.subheader("إضافة شكوى جديدة")
-        complaint_description = st.text_area("وصف الشكوى:")
+    df = load_data()
+
+    if st.session_state.role == "مقدم الشكوى":
+        st.header("📨 إرسال شكوى جديدة")
+        complaint_text = st.text_area("وصف الشكوى:")
         if st.button("إرسال الشكوى"):
-            if complaint_description:
-                new_complaint = {"مقدم الشكوى": username, "وصف الشكوى": complaint_description, "حالة الشكوى": "قيد المعالجة", "رد الشكوى": ""}
-                # استخدام pd.concat بدلاً من append
-                df = pd.concat([df, pd.DataFrame([new_complaint])], ignore_index=True)
-                df.to_csv(file_path, index=False)
-                st.success("تم إرسال الشكوى بنجاح.")
-            else:
-                st.error("الرجاء إدخال وصف الشكوى.")
+            user_id = st.session_state.username[-1] if st.session_state.username[-1].isdigit() else "X"
+            existing_ids = set(df["complaint_id"])
+            complaint_id = generate_complaint_id(user_id, existing_ids)
+            new_complaint = {
+                "complaint_id": complaint_id,
+                "username": st.session_state.username,
+                "complaint_text": complaint_text,
+                "status": "قيد الانتظار",
+                "response": "",
+                "timestamp": datetime.now().isoformat()
+            }
+            df = pd.concat([df, pd.DataFrame([new_complaint])], ignore_index=True)
+            save_data(df)
+            st.success(f"تم إرسال الشكوى بنجاح ✅\nرقم الشكوى الخاص بك: {complaint_id}")
 
-        # عرض الشكاوى المقدمة
-        st.subheader("الشكاوى المقدمة")
-        user_complaints = df[df['مقدم الشكوى'] == username]
-        if not user_complaints.empty:
-            st.write(user_complaints)
+        st.subheader("📋 الشكاوى الخاصة بك")
+        user_df = df[df["username"] == st.session_state.username]
+        if not user_df.empty:
+            for _, row in user_df.iterrows():
+                with st.expander(f"🔹 شكوى: {row['complaint_id']}"):
+                    st.write(f"**الحالة:** {row['status']}")
+                    st.write(f"**الرد:** {row['response'] if row['response'] else 'لم يتم الرد بعد'}")
+                    st.write(f"**الوصف:** {row['complaint_text']}")
         else:
-            st.write("لا توجد شكاوى مقدمة.")
+            st.info("لا توجد شكاوى بعد.")
 
-        # عرض حالة الشكوى
-        if not user_complaints.empty:
-            for idx, row in user_complaints.iterrows():
-                st.write(f"**حالة الشكوى:** {row['حالة الشكوى']}")
+    elif st.session_state.role == "مستقبل الشكوى":
+        st.header("📬 جميع الشكاوى")
+        if df.empty:
+            st.info("لا توجد شكاوى بعد")
         else:
-            st.write("لا توجد شكاوى مقدمة.")
-
-        # زر للخروج من النظام
-        if st.button("تسجيل الخروج"):
-            del st.session_state['user_type']
-            del st.session_state['username']
-            st.success("تم تسجيل الخروج بنجاح.")
-            st.experimental_rerun()
-
-    elif user_type == "مستقبل شكوى":
-        st.subheader("إدارة الشكاوى")
-        # عرض جميع الشكاوى للإداري للتعامل معها
-        if not df.empty:
-            for index, row in df.iterrows():
-                st.write(f"### شكوى من {row['مقدم الشكوى']}")
-                st.write(f"**الوصف:** {row['وصف الشكوى']}")
-                st.write(f"**الحالة الحالية:** {row['حالة الشكوى']}")
-                
-                # اختيار حالة الشكوى
-                new_status = st.selectbox(f"تغيير حالة الشكوى {row['مقدم الشكوى']}", ["قيد المعالجة", "مغلقة", "تم الرد عليها"], key=f"status_{index}")
-                if st.button(f"تحديث حالة الشكوى {row['مقدم الشكوى']}", key=f"update_{index}"):
-                    df.at[index, 'حالة الشكوى'] = new_status
-                    df.to_csv(file_path, index=False)
-                    st.success(f"تم تحديث حالة الشكوى لـ {row['مقدم الشكوى']}")
-
-                # إضافة رد على الشكوى
-                response = st.text_area(f"أضف ردًا على الشكوى من {row['مقدم الشكوى']}", key=f"response_{index}")
-                if st.button(f"إرسال الرد لـ {row['مقدم الشكوى']}", key=f"send_{index}"):
-                    df.at[index, 'رد الشكوى'] = response
-                    df.to_csv(file_path, index=False)
-                    st.success(f"تم إرسال الرد لـ {row['مقدم الشكوى']}")
+            for i, row in df.iterrows():
+                with st.expander(f"🧾 شكوى: {row['complaint_id']} (بواسطة {row['username']})"):
+                    st.write(f"**الوصف:** {row['complaint_text']}")
+                    st.write(f"**الحالة الحالية:** {row['status']}")
+                    new_status = st.selectbox("تغيير الحالة", ["قيد الانتظار", "جاري المعالجة", "تم الرد"], index=["قيد الانتظار", "جاري المعالجة", "تم الرد"].index(row['status']), key=f"status_{i}")
+                    response = st.text_area("الرد على الشكوى", value=row['response'], key=f"response_{i}")
+                    if st.button("تحديث", key=f"update_{i}"):
+                        df.at[i, 'status'] = new_status
+                        df.at[i, 'response'] = response
+                        save_data(df)
+                        st.success("تم تحديث الشكوى")