Update app.py

app.py

@@ -1,24 +1,99 @@
 import gradio as gr
 import pandas as pd
+import hashlib
+import json
+import os
+
+# Initialize admin credentials file if it doesn't exist
+ADMIN_FILE = "admin_credentials.json"
+if not os.path.exists(ADMIN_FILE):
+    default_admin = {
+        "admin": hashlib.sha256("admin123".encode()).hexdigest()
+    }
+    with open(ADMIN_FILE, "w") as f:
+        json.dump(default_admin, f)
+
+def load_admin_credentials():
+    with open(ADMIN_FILE, "r") as f:
+        return json.load(f)
+
+def save_admin_credentials(credentials):
+    with open(ADMIN_FILE, "w") as f:
+        json.dump(credentials, f)
 
 # Function to handle user feedback submission
 def submit_feedback(message):
-    # Append the message to the user feedback DataFrame    
-    user_feedback = pd.DataFrame({"message": [message]})    
-    user_feedback.to_csv("user_feedback.csv", mode='a', header=False, index=False)    
+    user_feedback = pd.DataFrame({"message": [message]})
+    user_feedback.to_csv("user_feedback.csv", mode='a', header=False, index=False)
     return "Feedback submitted successfully!"
 
-# Create a Gradio interface for submitting feedback
+# 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):
+    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:
+        return "Username already exists!"
+    
+    credentials[username] = hashlib.sha256(password.encode()).hexdigest()
+    save_admin_credentials(credentials)
+    return "New admin account 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
+    
+    try:
+        df = pd.read_csv("user_feedback.csv", names=["message"])
+        return df.to_string()
+    except:
+        return "No messages found or error reading messages."
+
+# Create Gradio interface
 with gr.Blocks() as demo:
-    # Textbox for user to input feedback    
-    feedback_input = gr.Textbox(label="Your Feedback")    
-    # Button to submit feedback    
-    submit_button = gr.Button("Submit Feedback")    
-    # Textbox to display the result of the submission    
-    feedback_output = gr.Textbox(label="Result")
+    gr.Markdown("# Feedback System")
+    
+    with gr.Tab("Submit Feedback"):
+        feedback_input = gr.Textbox(label="Your Feedback")
+        submit_button = gr.Button("Submit Feedback")
+        feedback_output = gr.Textbox(label="Result")
+        submit_button.click(fn=submit_feedback, inputs=feedback_input, outputs=feedback_output)
+    
+    with gr.Tab("Admin Login"):
+        login_username = gr.Textbox(label="Username")
+        login_password = gr.Textbox(label="Password", type="password")
+        view_messages_btn = gr.Button("View Messages")
+        messages_output = gr.Textbox(label="Messages", lines=10)
+        view_messages_btn.click(
+            fn=view_messages,
+            inputs=[login_username, login_password],
+            outputs=messages_output
+        )
     
-    # Set up the event listener for the submit button    
-    submit_button.click(fn=submit_feedback, inputs=feedback_input, outputs=feedback_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(
+            fn=create_admin,
+            inputs=[new_username, new_password, admin_password],
+            outputs=create_admin_output
+        )
 
 # Launch the Gradio app
 demo.launch(show_error=True)