Update app.py

app.py

@@ -15,8 +15,11 @@ if not os.path.exists(ADMIN_FILE):
         json.dump(default_admin, f)
 
 def load_admin_credentials():
-    with open(ADMIN_FILE, "r") as f:
-        return json.load(f)
+    try:
+        with open(ADMIN_FILE, "r") as f:
+            return json.load(f)
+    except:
+        return {"admin": hashlib.sha256("admin123".encode()).hexdigest()}
 
 def save_admin_credentials(credentials):
     with open(ADMIN_FILE, "w") as f:
@@ -26,143 +29,124 @@ def get_admin_usernames():
     credentials = load_admin_credentials()
     return list(credentials.keys())
 
+def refresh_admin_list():
+    return gr.Dropdown(choices=get_admin_usernames())
+
 # Function to handle user feedback submission
 def submit_feedback(admin_username, message):
+    if not admin_username or not message:
+        return "Please fill in both admin selection and 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 "Your feedback has been successfully submitted to the selected administrator."
+    
+    # Create file with headers if it doesn't exist
+    if not os.path.exists("user_feedback.csv"):
+        user_feedback.to_csv("user_feedback.csv", index=False)
+    else:
+        user_feedback.to_csv("user_feedback.csv", mode='a', header=False, index=False)
+    
+    return "Feedback submitted successfully!"
 
 # Function to create new admin account
-def create_admin(new_username, new_password):
-    if not new_username or not new_password:
+def create_admin(username, password):
+    if not username or not password:
         return "Please fill in both username and password fields."
     
-    credentials = load_admin_credentials()
-    if new_username in credentials:
-        return "This username is already taken. Please choose a different username."
+    try:
+        credentials = load_admin_credentials()
+        
+        if username in credentials:
+            return "Username already exists. Please choose a different username."
+        
+        credentials[username] = hashlib.sha256(password.encode()).hexdigest()
+        save_admin_credentials(credentials)
+        return "Account created successfully! You can now log in with your credentials."
     
-    credentials[new_username] = hashlib.sha256(new_password.encode()).hexdigest()
-    save_admin_credentials(credentials)
-    return f"Success! New administrator account for '{new_username}' has been created. You can now log in using these credentials."
+    except Exception as e:
+        return f"Error creating account: {str(e)}"
 
 # Function to view feedback messages
 def view_messages(username, password):
-    credentials = load_admin_credentials()
-    hashed_password = hashlib.sha256(password.encode()).hexdigest()
-    if username not in credentials or credentials[username] != hashed_password:
-        return "Login failed. Please check your username and password and try again."
+    if not username or not password:
+        return "Please provide both username and password."
     
     try:
-        df = pd.read_csv("user_feedback.csv", names=["timestamp", "admin", "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. Please try again."
+        
+        if not os.path.exists("user_feedback.csv"):
+            return "No messages found."
+            
+        df = pd.read_csv("user_feedback.csv")
         admin_messages = df[df['admin'] == username]
+        
         if admin_messages.empty:
-            return "You have no messages at this time."
+            return "No messages found for your account."
+            
         return admin_messages.to_string()
-    except FileNotFoundError:
-        return "No messages found in the system."
+        
     except Exception as e:
-        return f"An error occurred while retrieving messages: {str(e)}"
+        return f"Error retrieving messages: {str(e)}"
 
 # Create Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Feedback Management System")
-    gr.Markdown("Welcome to the Feedback Management System. Please select a tab below to continue.")
+    gr.Markdown("# Feedback System")
     
-    with gr.Tab("Submit Feedback", elem_id="submit_tab"):
-        gr.Markdown("### Submit Your Feedback")
-        gr.Markdown("Please select an administrator and enter your feedback message below.")
+    with gr.Tab("Submit Feedback"):
         admin_select = gr.Dropdown(
             choices=get_admin_usernames(),
-            label="Select Administrator",
-            info="Choose the administrator who should receive your feedback",
-            elem_id="admin_select"
+            label="Select Admin",
+            interactive=True
         )
         feedback_input = gr.Textbox(
-            label="Your Feedback Message",
-            lines=3,
-            placeholder="Type your feedback here",
-            info="Enter your feedback message in detail",
-            elem_id="feedback_input"
-        )
-        submit_button = gr.Button("Submit Feedback", elem_id="submit_button")
-        feedback_output = gr.Textbox(
-            label="Submission Status",
-            interactive=False,
-            elem_id="feedback_status"
+            label="Your Feedback",
+            placeholder="Enter your feedback here",
+            lines=3
         )
+        submit_button = gr.Button("Submit")
+        feedback_output = gr.Textbox(label="Status")
+        
         submit_button.click(
-            fn=submit_feedback, 
-            inputs=[admin_select, feedback_input], 
+            fn=submit_feedback,
+            inputs=[admin_select, feedback_input],
             outputs=feedback_output
         )
     
-    with gr.Tab("Administrator Login", elem_id="login_tab"):
-        gr.Markdown("### Administrator Login")
-        gr.Markdown("Please log in with your administrator credentials to view messages.")
-        login_username = gr.Dropdown(
-            choices=get_admin_usernames(),
-            label="Administrator Username",
-            info="Select your username from the list",
-            elem_id="login_username"
-        )
-        login_password = gr.Textbox(
-            label="Password",
-            type="password",
-            placeholder="Enter your password",
-            info="Enter your administrator password",
-            elem_id="login_password"
-        )
-        view_messages_btn = gr.Button("View Messages", elem_id="view_messages_button")
-        messages_output = gr.Textbox(
-            label="Messages",
-            lines=10,
-            interactive=False,
-            elem_id="messages_display"
-        )
-        view_messages_btn.click(
+    with gr.Tab("Admin Login"):
+        login_username = gr.Textbox(label="Username")
+        login_password = gr.Textbox(label="Password", type="password")
+        view_button = gr.Button("View Messages")
+        messages_output = gr.Textbox(label="Messages", lines=10)
+        
+        view_button.click(
             fn=view_messages,
             inputs=[login_username, login_password],
             outputs=messages_output
         )
     
-    with gr.Tab("Create Administrator Account", elem_id="create_account_tab"):
-        gr.Markdown("### Create New Administrator Account")
-        gr.Markdown("Fill in the fields below to create a new administrator account.")
-        new_username = gr.Textbox(
-            label="New Administrator Username",
-            placeholder="Enter desired username",
-            info="Choose a unique username for the new administrator account",
-            elem_id="new_username"
-        )
-        new_password = gr.Textbox(
-            label="New Administrator Password",
-            type="password",
-            placeholder="Enter desired password",
-            info="Choose a secure password for the account",
-            elem_id="new_password"
-        )
-        create_button = gr.Button(
-            "Create Administrator Account",
-            elem_id="create_account_button"
-        )
-        create_output = gr.Textbox(
-            label="Account Creation Status",
-            interactive=False,
-            elem_id="creation_status"
-        )
+    with gr.Tab("Create Account"):
+        new_username = gr.Textbox(label="Username")
+        new_password = gr.Textbox(label="Password", type="password")
+        create_button = gr.Button("Create Account")
+        create_output = gr.Textbox(label="Status")
+        
+        # Create account and update admin lists
         create_button.click(
             fn=create_admin,
             inputs=[new_username, new_password],
             outputs=create_output
+        ).then(
+            fn=refresh_admin_list,
+            inputs=None,
+            outputs=admin_select
         )
 
-    gr.Markdown("### Need Help?")
-    gr.Markdown("If you need assistance, please contact system support.")
-
-# Launch the Gradio app
-demo.launch(show_error=True)
+demo.launch(share=True)