Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import hashlib
 import json
 import os
+from datetime import datetime
 
 # Initialize admin credentials file if it doesn't exist
 ADMIN_FILE = "admin_credentials.json"
@@ -21,59 +22,66 @@ def save_admin_credentials(credentials):
     with open(ADMIN_FILE, "w") as f:
         json.dump(credentials, f)
 
+def get_admin_usernames():
+    credentials = load_admin_credentials()
+    return list(credentials.keys())
+
 # Function to handle user feedback submission
-def submit_feedback(message):
-    user_feedback = pd.DataFrame({"message": [message]})
+def submit_feedback(admin_username, message):
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    user_feedback = pd.DataFrame({
+        "timestamp": [timestamp],
+        "admin": [admin_username],
+        "message": [message]
+    })
     user_feedback.to_csv("user_feedback.csv", mode='a', header=False, index=False)
     return "Feedback submitted successfully!"
 
-# Function to verify admin login
-def verify_login(username, password):
-    credentials = load_admin_credentials()
-    hashed_password = hashlib.sha256(password.encode()).hexdigest()
-    if username in credentials and credentials[username] == hashed_password:
-        return True, "Login successful!"
-    return False, "Invalid credentials!"
-
 # Function to create new admin account
-def create_admin(username, password, admin_password):
+def create_admin(new_username, new_password):
     credentials = load_admin_credentials()
-    
-    # Verify current admin password
-    if hashlib.sha256(admin_password.encode()).hexdigest() not in credentials.values():
-        return "Invalid admin password!"
-    
-    if username in credentials:
+    if new_username in credentials:
         return "Username already exists!"
     
-    credentials[username] = hashlib.sha256(password.encode()).hexdigest()
+    credentials[new_username] = hashlib.sha256(new_password.encode()).hexdigest()
     save_admin_credentials(credentials)
-    return "New admin account created successfully!"
+    return f"New admin account '{new_username}' created successfully!"
 
 # Function to view feedback messages
 def view_messages(username, password):
-    login_success, message = verify_login(username, password)
-    if not login_success:
-        return message
+    credentials = load_admin_credentials()
+    hashed_password = hashlib.sha256(password.encode()).hexdigest()
+    if username not in credentials or credentials[username] != hashed_password:
+        return "Invalid credentials!"
     
     try:
-        df = pd.read_csv("user_feedback.csv", names=["message"])
-        return df.to_string()
-    except:
-        return "No messages found or error reading messages."
+        df = pd.read_csv("user_feedback.csv", names=["timestamp", "admin", "message"])
+        admin_messages = df[df['admin'] == username]
+        if admin_messages.empty:
+            return "No messages found for your account."
+        return admin_messages.to_string()
+    except FileNotFoundError:
+        return "No messages found."
+    except Exception as e:
+        return f"Error reading messages: {str(e)}"
 
 # Create Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Feedback System")
     
     with gr.Tab("Submit Feedback"):
-        feedback_input = gr.Textbox(label="Your Feedback")
+        admin_select = gr.Dropdown(choices=get_admin_usernames(), label="Select Admin")
+        feedback_input = gr.Textbox(label="Your Feedback", lines=3)
         submit_button = gr.Button("Submit Feedback")
         feedback_output = gr.Textbox(label="Result")
-        submit_button.click(fn=submit_feedback, inputs=feedback_input, outputs=feedback_output)
+        submit_button.click(
+            fn=submit_feedback, 
+            inputs=[admin_select, feedback_input], 
+            outputs=feedback_output
+        )
     
     with gr.Tab("Admin Login"):
-        login_username = gr.Textbox(label="Username")
+        login_username = gr.Dropdown(choices=get_admin_usernames(), label="Admin Username")
         login_password = gr.Textbox(label="Password", type="password")
         view_messages_btn = gr.Button("View Messages")
         messages_output = gr.Textbox(label="Messages", lines=10)
@@ -83,16 +91,15 @@ with gr.Blocks() as demo:
             outputs=messages_output
         )
     
-    with gr.Tab("Create Admin Account"):
-        new_username = gr.Textbox(label="New Username")
-        new_password = gr.Textbox(label="New Password", type="password")
-        admin_password = gr.Textbox(label="Current Admin Password", type="password")
-        create_admin_btn = gr.Button("Create Admin Account")
-        create_admin_output = gr.Textbox(label="Result")
-        create_admin_btn.click(
+    with gr.Tab("Create New Admin"):
+        new_username = gr.Textbox(label="New Admin Username")
+        new_password = gr.Textbox(label="New Admin Password", type="password")
+        create_button = gr.Button("Create New Admin")
+        create_output = gr.Textbox(label="Result")
+        create_button.click(
             fn=create_admin,
-            inputs=[new_username, new_password, admin_password],
-            outputs=create_admin_output
+            inputs=[new_username, new_password],
+            outputs=create_output
         )
 
 # Launch the Gradio app