Update app.py

app.py

@@ -1,112 +1,131 @@
 import streamlit as st
 import pandas as pd
 import os
-import uuid
 from datetime import datetime
 
-st.set_page_config(page_title="نظام الشكاوى", layout="centered")
+st.set_page_config(page_title="نظام إدارة الشكاوى", layout="centered")
 
-# اسم الملف لتخزين الشكاوى
-csv_file = "complaints.csv"
-
-# التأكد من وجود الملف
-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
+# -----------------------------------------
+# تحميل أو إنشاء ملف CSV
+file_path = "complaints.csv"
+if os.path.exists(file_path):
+    df = pd.read_csv(file_path)
+else:
+    df = pd.DataFrame(columns=["complaint_id", "username", "complaint_text", "status", "response", "timestamp"])
+    df.to_csv(file_path, index=False)
 
-# جلسة المستخدم
-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
+# -----------------------------------------
+# تهيئة session state
+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 "last_seen_count" not in st.session_state:
+    st.session_state.last_seen_count = len(df)
 
+# -----------------------------------------
 # تسجيل الدخول
-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 not st.session_state.is_logged_in:
+    st.title("🔐 Sign In")
+    username = st.text_input("User Name ")
+    password = st.text_input("Password", type="password")
+    if st.button("Sing in "):
         if username and password:
-            st.session_state.logged_in = True
-            st.session_state.role = role
+            if username.startswith("admin"):
+                st.session_state.user_type = "receiver"
+            else:
+                st.session_state.user_type = "sender"
+            st.session_state.is_logged_in = True
             st.session_state.username = username
             st.rerun()
         else:
-            st.error("من فضلك أدخل اسم المستخدم وكلمة المرور")
+            st.warning("يرجى إدخال اسم المستخدم وكلمة المرور")
 
-else:
-    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 = ""
+# -----------------------------------------
+# مقدم الشكوى
+elif st.session_state.user_type == "sender":
+    st.title("📨 تقديم شكوى")
+    complaint_text = st.text_area("Description ")
+    if st.button("Send complaint"):
+        user_id = st.session_state.username.split("_")[-1].upper()
+        existing_ids = df[df["username"] == st.session_state.username]["complaint_id"]
+        next_id = len(existing_ids) + 1
+        complaint_id = f"SUE {user_id}-{next_id:06}"
+        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        new_complaint = {
+            "complaint_id": complaint_id,
+            "username": st.session_state.username,
+            "complaint_text": complaint_text,
+            "status": "Pending",
+            "response": "",
+            "timestamp": timestamp
+        }
+        df = pd.concat([df, pd.DataFrame([new_complaint])], ignore_index=True)
+        df.to_csv(file_path, index=False)
+        st.success("✅ تم تقديم الشكوى بنجاح")
         st.rerun()
 
-    df = load_data()
+    # إشعار بتحديث حالة الشكوى أو الرد
+    user_complaints = df[df["username"] == st.session_state.username]
+    if not user_complaints.empty:
+        last_complaint = user_complaints.iloc[-1]
+        if last_complaint["status"].lower() != "pending" or pd.notna(last_complaint["response"]):
+            st.info(f"✅ تم تحديث حالة شكواك ({last_complaint['complaint_id']}) إلى: {last_complaint['status']}")
+            if pd.notna(last_complaint["response"]) and last_complaint["response"].strip() != "":
+                st.success(f"💬 تم الرد على شكوا��: {last_complaint['response']}")
 
-    if st.session_state.role == "مقدم الشكوى":
-        st.header("📨 إرسال شكوى جديدة")
-        complaint_text = st.text_area("وصف الشكوى:")
-        if st.button("إرسال الشكوى"):
-            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("📋 last complaints")
+    st.dataframe(user_complaints[["complaint_id", "complaint_text", "status", "response", "timestamp"]], use_container_width=True)
 
-        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.info("لا توجد شكاوى بعد.")
+    # زر تسجيل الخروج
+    if st.button("🚪 Log out"):
+        st.session_state.is_logged_in = False
+        st.session_state.user_type = None
+        st.rerun()
 
-    elif st.session_state.role == "مستقبل الشكوى":
-        st.header("📬 جميع الشكاوى")
-        if df.empty:
-            st.info("لا توجد شكاوى بعد")
-        else:
-            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("تم تحديث الشكوى")
+# -----------------------------------------
+# مستقبل الشكوى
+elif st.session_state.user_type == "receiver":
+    st.title("📬 إدارة الشكاوى")
+
+    # تنبيه عن شكاوى جديدة
+    current_count = len(df)
+    new_complaints = current_count - st.session_state.last_seen_count
+    if new_complaints > 0:
+        st.success(f"📬 تم تقديم {new_complaints} شكوى جديدة منذ آخر زيارة.")
+    st.session_state.last_seen_count = current_count
+
+    # عرض الشكاوى
+    st.subheader("📋 جميع الشكاوى")
+    selected_index = st.selectbox("اختر شكوى للتعامل معها", df.index, format_func=lambda i: f"{df.loc[i, 'complaint_id']} - {df.loc[i, 'username']}")
+    selected_row = df.loc[selected_index]
+    st.write(f"### ✉️ شكوى: {selected_row['complaint_id']}")
+    st.write(f"**مقدم الشكوى:** {selected_row['username']}")
+    st.write(f"**الوصف:** {selected_row['complaint_text']}")
+
+    new_status = st.selectbox("🔄 تحديث الحالة", ["Pending", "In Progress", "Resolved"], index=["Pending", "In Progress", "Resolved"].index(selected_row["status"]))
+    new_response = st.text_area("📝 الرد على الشكوى", value=selected_row["response"] or "")
+
+    if st.button("💾 حفظ التعديلات"):
+        df.at[selected_index, "status"] = new_status
+        df.at[selected_index, "response"] = new_response
+        df.to_csv(file_path, index=False)
+        st.success("✅ تم حفظ التعديلات")
+        st.rerun()
+
+    # زر لتحميل الشكاوى
+    st.download_button(
+        label="📥 تحميل الشكاوى كـ CSV",
+        data=df.to_csv(index=False).encode('utf-8'),
+        file_name="complaints_export.csv",
+        mime="text/csv"
+    )
+
+    # زر تسجيل الخروج
+    if st.button("🚪Sign Out"):
+        st.session_state.is_logged_in = False
+        st.session_state.user_type = None
+        st.rerun()