Update app.py

app.py

@@ -1,185 +1,201 @@
-import os
-import tkinter as tk
-from tkinter import messagebox, filedialog, scrolledtext
-import platform
-import webbrowser
-
-# ---------------- LOGIN SCREEN ---------------- #
-
-class LoginScreen:
-    def __init__(self, root):
-        self.root = root
-        self.root.title("Mini OS - Login")
-        self.root.geometry("400x300")
-        self.root.config(bg="black")
-
-        tk.Label(root, text="Mini Python OS", fg="white", bg="black",
-                 font=("Arial", 20)).pack(pady=20)
-
-        tk.Label(root, text="Username", bg="black", fg="white").pack()
-        self.username = tk.Entry(root)
-        self.username.pack()
-
-        tk.Label(root, text="Password", bg="black", fg="white").pack()
-        self.password = tk.Entry(root, show="*")
-        self.password.pack()
-
-        tk.Button(root, text="Login", command=self.login,
-                  bg="green", fg="white").pack(pady=20)
-
-    def login(self):
-        if self.username.get() == "admin" and self.password.get() == "1234":
-            self.root.destroy()
-            main_os()
-        else:
-            messagebox.showerror("Error", "Invalid Login")
-
-
-# ---------------- MAIN OS ---------------- #
-
-class MiniOS:
-    def __init__(self, root):
-        self.root = root
-        self.root.title("Mini Python OS")
-        self.root.geometry("800x500")
-        self.root.config(bg="#1e1e1e")
-
-        tk.Label(root, text="Mini Python OS Desktop",
-                 bg="#1e1e1e", fg="white",
-                 font=("Arial", 18)).pack(pady=20)
-
-        tk.Button(root, text="File Explorer",
-                  command=self.file_explorer,
-                  width=25, height=2).pack(pady=5)
-
-        tk.Button(root, text="Notepad",
-                  command=self.notepad,
-                  width=25, height=2).pack(pady=5)
-
-        tk.Button(root, text="Calculator",
-                  command=self.calculator,
-                  width=25, height=2).pack(pady=5)
-
-        tk.Button(root, text="System Info",
-                  command=self.system_info,
-                  width=25, height=2).pack(pady=5)
-
-        tk.Button(root, text="Open Google",
-                  command=self.open_google,
-                  width=25, height=2).pack(pady=5)
-
-        tk.Button(root, text="Open ChatGPT",
-                  command=self.open_chatgpt,
-                  width=25, height=2).pack(pady=5)
-
-        tk.Button(root, text="Shutdown",
-                  command=self.shutdown,
-                  width=25, height=2,
-                  bg="red", fg="white").pack(pady=20)
-
-    # -------- File Explorer -------- #
-    def file_explorer(self):
-        path = filedialog.askdirectory()
-        if path:
-            files = os.listdir(path)
-            messagebox.showinfo("Files", "\n".join(files))
-
-    # -------- Notepad -------- #
-    def notepad(self):
-        pad = tk.Toplevel(self.root)
-        pad.title("Notepad")
-        pad.geometry("600x400")
-
-        text_area = scrolledtext.ScrolledText(pad)
-        text_area.pack(fill="both", expand=True)
-
-        def save_file():
-            file_path = filedialog.asksaveasfilename(defaultextension=".txt")
-            if file_path:
-                with open(file_path, "w") as f:
-                    f.write(text_area.get("1.0", tk.END))
-                messagebox.showinfo("Saved", "File Saved Successfully")
-
-        tk.Button(pad, text="Save", command=save_file).pack()
-
-    # -------- Calculator -------- #
-    def calculator(self):
-        calc = tk.Toplevel(self.root)
-        calc.title("Calculator")
-        calc.geometry("300x400")
-
-        entry = tk.Entry(calc, font=("Arial", 18))
-        entry.pack(fill="both")
-
-        def click(value):
-            entry.insert(tk.END, value)
-
-        def calculate():
-            try:
-                result = eval(entry.get())
-                entry.delete(0, tk.END)
-                entry.insert(0, result)
-            except:
-                messagebox.showerror("Error", "Invalid Expression")
-
-        buttons = [
-            '7','8','9','/',
-            '4','5','6','*',
-            '1','2','3','-',
-            '0','.','=','+'
-        ]
-
-        frame = tk.Frame(calc)
-        frame.pack()
-
-        row = 0
-        col = 0
-
-        for button in buttons:
-            action = calculate if button == '=' else lambda x=button: click(x)
-            tk.Button(frame, text=button, width=5, height=2,
-                      command=action).grid(row=row, column=col)
-            col += 1
-            if col > 3:
-                col = 0
-                row += 1
-
-    # -------- System Info -------- #
-    def system_info(self):
-        info = f"""
-System: {platform.system()}
-Node Name: {platform.node()}
-Release: {platform.release()}
-Version: {platform.version()}
-Machine: {platform.machine()}
-Processor: {platform.processor()}
-        """
-        messagebox.showinfo("System Info", info)
-
-    # -------- Google -------- #
-    def open_google(self):
-        webbrowser.open("https://www.google.com")
-
-    # -------- ChatGPT -------- #
-    def open_chatgpt(self):
-        webbrowser.open("https://chat.openai.com")
-
-    # -------- Shutdown -------- #
-    def shutdown(self):
-        self.root.destroy()
-
-
-# -------- START MAIN OS -------- #
-
-def main_os():
-    root = tk.Tk()
-    app = MiniOS(root)
-    root.mainloop()
-
-
-# -------- RUN LOGIN FIRST -------- #
-
-if __name__ == "__main__":
-    root = tk.Tk()
-    LoginScreen(root)
-    root.mainloop()
+from flask import Flask, render_template_string, request, redirect, url_for, session
+import os
+import platform
+
+app = Flask(__name__)
+app.secret_key = "mini_os_secret_key"
+
+# ---------------- LOGIN PAGE ---------------- #
+login_page = """
+<!DOCTYPE html>
+<html>
+<head>
+    <title>Mini OS Login</title>
+    <style>
+        body { background:black; color:white; text-align:center; font-family:Arial; }
+        .box { margin-top:100px; }
+        input { padding:10px; margin:5px; width:200px; }
+        button { padding:10px 20px; background:green; color:white; border:none; }
+    </style>
+</head>
+<body>
+    <div class="box">
+        <h1>Mini Python OS</h1>
+        <form method="POST">
+            <input name="username" placeholder="Username"><br>
+            <input name="password" type="password" placeholder="Password"><br>
+            <button type="submit">Login</button>
+        </form>
+        <p>Default: admin / 1234</p>
+        <p style="color:red">{{error}}</p>
+    </div>
+</body>
+</html>
+"""
+
+# ---------------- DASHBOARD ---------------- #
+dashboard_page = """
+<!DOCTYPE html>
+<html>
+<head>
+    <title>Mini OS Desktop</title>
+    <style>
+        body { background:#1e1e1e; color:white; font-family:Arial; text-align:center; }
+        button { padding:15px; margin:10px; width:200px; cursor:pointer; }
+        .container { margin-top:50px; }
+    </style>
+</head>
+<body>
+    <h1>Mini Python OS Desktop</h1>
+    <div class="container">
+
+        <a href="/files"><button>File Explorer</button></a>
+        <a href="/notepad"><button>Notepad</button></a>
+        <a href="/calculator"><button>Calculator</button></a>
+        <a href="/system"><button>System Info</button></a>
+        <a href="https://google.com" target="_blank"><button>Open Google</button></a>
+        <a href="https://chat.openai.com" target="_blank"><button>Open ChatGPT</button></a>
+
+        <br><br>
+        <a href="/logout"><button style="background:red;color:white;">Logout</button></a>
+    </div>
+</body>
+</html>
+"""
+
+# ---------------- FILE EXPLORER ---------------- #
+files_page = """
+<h2>File Explorer</h2>
+<ul>
+{% for f in files %}
+<li>{{f}}</li>
+{% endfor %}
+</ul>
+<a href="/dashboard">Back</a>
+"""
+
+# ---------------- NOTEPAD ---------------- #
+notepad_page = """
+<h2>Notepad</h2>
+<form method="POST">
+<textarea name="text" rows="15" cols="50"></textarea><br>
+<button type="submit">Save</button>
+</form>
+<p>{{msg}}</p>
+<a href="/dashboard">Back</a>
+"""
+
+# ---------------- CALCULATOR ---------------- #
+calculator_page = """
+<h2>Calculator</h2>
+<form method="POST">
+<input name="exp" placeholder="Enter expression"><br><br>
+<button type="submit">Calculate</button>
+</form>
+<h3>Result: {{result}}</h3>
+<a href="/dashboard">Back</a>
+"""
+
+# ---------------- SYSTEM INFO ---------------- #
+system_page = """
+<h2>System Info</h2>
+<pre>
+System: {{system}}
+Node: {{node}}
+Release: {{release}}
+Version: {{version}}
+Machine: {{machine}}
+Processor: {{processor}}
+</pre>
+<a href="/dashboard">Back</a>
+"""
+
+# ---------------- ROUTES ---------------- #
+
+@app.route("/", methods=["GET", "POST"])
+def login():
+    error = ""
+    if request.method == "POST":
+        u = request.form["username"]
+        p = request.form["password"]
+
+        if u == "admin" and p == "1234":
+            session["user"] = u
+            return redirect("/dashboard")
+        else:
+            error = "Invalid Login"
+
+    return render_template_string(login_page, error=error)
+
+
+@app.route("/dashboard")
+def dashboard():
+    if "user" not in session:
+        return redirect("/")
+    return render_template_string(dashboard_page)
+
+
+@app.route("/files")
+def files():
+    if "user" not in session:
+        return redirect("/")
+    path = "."
+    file_list = os.listdir(path)
+    return render_template_string(files_page, files=file_list)
+
+
+@app.route("/notepad", methods=["GET", "POST"])
+def notepad():
+    if "user" not in session:
+        return redirect("/")
+
+    msg = ""
+    if request.method == "POST":
+        text = request.form["text"]
+        with open("note.txt", "w") as f:
+            f.write(text)
+        msg = "Saved to note.txt"
+
+    return render_template_string(notepad_page, msg=msg)
+
+
+@app.route("/calculator", methods=["GET", "POST"])
+def calculator():
+    if "user" not in session:
+        return redirect("/")
+
+    result = ""
+    if request.method == "POST":
+        try:
+            exp = request.form["exp"]
+            result = eval(exp)
+        except:
+            result = "Error"
+
+    return render_template_string(calculator_page, result=result)
+
+
+@app.route("/system")
+def system():
+    if "user" not in session:
+        return redirect("/")
+
+    return render_template_string(system_page,
+        system=platform.system(),
+        node=platform.node(),
+        release=platform.release(),
+        version=platform.version(),
+        machine=platform.machine(),
+        processor=platform.processor()
+    )
+
+
+@app.route("/logout")
+def logout():
+    session.clear()
+    return redirect("/")
+
+
+# ---------------- RUN APP ---------------- #
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)