Update app.py

app.py

@@ -1,73 +1,88 @@
 import streamlit as st
-import os
-import shutil
-from git import Repo
+import sqlite3
+import hashlib
 
-# Set up a directory for saving files
-SAVE_DIR = "saved_projects"
+# Database setup
+DB_FILE = "users.db"
 
-if not os.path.exists(SAVE_DIR):
-    os.makedirs(SAVE_DIR)
+def create_user_table():
+    conn = sqlite3.connect(DB_FILE)
+    cursor = conn.cursor()
+    cursor.execute("""
+        CREATE TABLE IF NOT EXISTS users (
+            username TEXT PRIMARY KEY,
+            password TEXT
+        )
+    """)
+    conn.commit()
+    conn.close()
 
-def save_uploaded_files(uploaded_files):
-    """Save uploaded files to the SAVE_DIR."""
-    project_dir = os.path.join(SAVE_DIR, "uploaded_project")
-    if os.path.exists(project_dir):
-        shutil.rmtree(project_dir)  # Clean up existing files
-    os.makedirs(project_dir)
-    
-    for uploaded_file in uploaded_files:
-        with open(os.path.join(project_dir, uploaded_file.name), "wb") as f:
-            f.write(uploaded_file.getbuffer())
-    return project_dir
+def add_user(username, password):
+    conn = sqlite3.connect(DB_FILE)
+    cursor = conn.cursor()
+    hashed_password = hashlib.sha256(password.encode()).hexdigest()
+    try:
+        cursor.execute("INSERT INTO users (username, password) VALUES (?, ?)", (username, hashed_password))
+        conn.commit()
+    except sqlite3.IntegrityError:
+        st.error("Username already exists. Please choose a different username.")
+    conn.close()
 
-def clone_repository(repo_url):
-    """Clone a GitHub repository into SAVE_DIR."""
-    repo_name = repo_url.split("/")[-1].replace(".git", "")
-    repo_dir = os.path.join(SAVE_DIR, repo_name)
-    if os.path.exists(repo_dir):
-        shutil.rmtree(repo_dir)  # Clean up existing files
-    Repo.clone_from(repo_url, repo_dir)
-    return repo_dir
+def authenticate_user(username, password):
+    conn = sqlite3.connect(DB_FILE)
+    cursor = conn.cursor()
+    hashed_password = hashlib.sha256(password.encode()).hexdigest()
+    cursor.execute("SELECT * FROM users WHERE username = ? AND password = ?", (username, hashed_password))
+    user = cursor.fetchone()
+    conn.close()
+    return user
 
+# Main application
 def main():
-    st.title("Project Manager App")
-    st.write("Upload your project files or clone a GitHub repository.")
+    st.title("Welcome to Your Streamlit App")
 
-    # Sidebar for navigation
-    option = st.sidebar.selectbox(
-        "Choose an option:",
-        ("Upload Files", "Clone GitHub Repository")
-    )
+    # Initialize database
+    create_user_table()
 
-    if option == "Upload Files":
-        st.header("Upload Files")
-        uploaded_files = st.file_uploader(
-            "Upload your coding project files (multiple files allowed):",
-            accept_multiple_files=True
-        )
-        if uploaded_files:
-            saved_path = save_uploaded_files(uploaded_files)
-            st.success(f"Files saved to {saved_path}")
+    # Authentication state
+    if "authenticated" not in st.session_state:
+        st.session_state.authenticated = False
+
+    # Central login/register interface
+    if not st.session_state.authenticated:
+        st.subheader("Please Log In or Register to Continue")
+        auth_mode = st.radio("Choose an Option", ["Log In", "Register"], horizontal=True)
+
+        if auth_mode == "Log In":
+            st.subheader("Log In")
+            username = st.text_input("Username", key="login_username")
+            password = st.text_input("Password", type="password", key="login_password")
 
-    elif option == "Clone GitHub Repository":
-        st.header("Clone GitHub Repository")
-        repo_url = st.text_input("Enter the GitHub repository URL:")
-        if repo_url:
-            if st.button("Clone Repository"):
-                try:
-                    repo_path = clone_repository(repo_url)
-                    st.success(f"Repository cloned to {repo_path}")
-                except Exception as e:
-                    st.error(f"Error cloning repository: {e}")
+            if st.button("Log In"):
+                if authenticate_user(username, password):
+                    st.success(f"Welcome back, {username}!")
+                    st.session_state.authenticated = True
+                    st.session_state.username = username
+                else:
+                    st.error("Invalid username or password. Please try again.")
 
-    # Display saved projects
-    st.header("Saved Projects")
-    if os.listdir(SAVE_DIR):
-        for project in os.listdir(SAVE_DIR):
-            st.write(f"- {project}")
-    else:
-        st.write("No projects saved yet.")
+        elif auth_mode == "Register":
+            st.subheader("Register")
+            username = st.text_input("Create Username", key="register_username")
+            password = st.text_input("Create Password", type="password", key="register_password")
+
+            if st.button("Register"):
+                if username and password:
+                    add_user(username, password)
+                    st.success("Account created successfully! You can now log in.")
+                else:
+                    st.error("Please fill in all fields.")
+
+    # Authenticated user workspace
+    if st.session_state.authenticated:
+        st.subheader(f"Hello, {st.session_state.username}!")
+        st.write("This is your workspace. All your saved work will appear here.")
 
 if __name__ == "__main__":
     main()
+