Update app.py

app.py

@@ -2,18 +2,12 @@ import os
 import json
 import bcrypt
 import gradio as gr
-from flask import Flask, request, jsonify
 from flask_jwt_extended import (
     JWTManager, create_access_token,
     jwt_required, get_jwt_identity
 )
 
-# Flask setup
-flask_app = Flask(__name__)
-flask_app.config["JWT_SECRET_KEY"] = os.getenv("JWT_SECRET_KEY", "super-secret-key")
-jwt = JWTManager(flask_app)
-
-# File-based user storage
+# Gradio app will handle everything
 USERS_FILE = "users.json"
 
 def load_users():
@@ -27,90 +21,46 @@ def save_users(users):
     with open(USERS_FILE, "w") as f:
         json.dump(users, f)
 
-# ============================
-# Flask API routes
-# ============================
-
-@flask_app.route("/register", methods=["POST"])
-def register():
-    data = request.get_json()
-    username = data.get("username")
-    password = data.get("password")
+# JWT setup (simplified for Gradio)
+JWT_SECRET_KEY = os.getenv("JWT_SECRET_KEY", "super-secret-key")
 
+def register_user(username, password):
     if not username or not password:
-        return jsonify({"error": "Username and password required"}), 400
-
+        return "Username and password required"
+    
     users = load_users()
     if username in users:
-        return jsonify({"error": "Username already exists"}), 400
-
+        return "Username already exists"
+    
     hashed = bcrypt.hashpw(password.encode(), bcrypt.gensalt()).decode()
     users[username] = {"password": hashed}
     save_users(users)
+    return "User registered successfully"
 
-    return jsonify({"message": "User registered successfully"}), 201
-
-@flask_app.route("/login", methods=["POST"])
-def login():
-    data = request.get_json()
-    username = data.get("username")
-    password = data.get("password")
-
+def login_user(username, password):
     users = load_users()
     user = users.get(username)
-
+    
     if not user or not bcrypt.checkpw(password.encode(), user["password"].encode()):
-        return jsonify({"error": "Invalid credentials"}), 401
-
-    access_token = create_access_token(identity=username)
-    return jsonify({"access_token": access_token, "username": username}), 200
-
-@flask_app.route("/protected", methods=["GET"])
-@jwt_required()
-def protected():
-    current_user = get_jwt_identity()
-    return jsonify(logged_in_as=current_user), 200
-
-# ============================
-# Gradio UI
-# ============================
-
-# For local development, use local Flask server
-LOCAL_FLASK_URL = "http://localhost:5000"
-
-def register_interface(username, password):
-    try:
-        response = requests.post(
-            f"{LOCAL_FLASK_URL}/register",
-            json={"username": username, "password": password}
-        )
-        return response.json().get("message", response.json().get("error", "Unknown error"))
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-def login_interface(username, password):
-    try:
-        response = requests.post(
-            f"{LOCAL_FLASK_URL}/login",
-            json={"username": username, "password": password}
-        )
-        if response.status_code == 200:
-            token = response.json().get("access_token")
-            return f"Login successful! Token: {token[:15]}... (truncated)"
-        return response.json().get("error", "Login failed")
-    except Exception as e:
-        return f"Error: {str(e)}"
+        return "Invalid credentials", None
+    
+    access_token = create_access_token(identity=username, secret_key=JWT_SECRET_KEY)
+    return "Login successful", access_token
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 🔐 Secure Auth System (Gradio + Flask + JWT + bcrypt)")
+    gr.Markdown("# 🔐 Secure Auth System")
     
     with gr.Tab("Register"):
         reg_username = gr.Textbox(label="Username")
         reg_password = gr.Textbox(label="Password", type="password")
         reg_output = gr.Textbox(label="Output")
         reg_button = gr.Button("Register")
+        
+        def register(username, password):
+            return register_user(username, password)
+            
         reg_button.click(
-            register_interface,
+            register,
             inputs=[reg_username, reg_password],
             outputs=reg_output
         )
@@ -120,25 +70,17 @@ with gr.Blocks() as demo:
         login_password = gr.Textbox(label="Password", type="password")
         login_output = gr.Textbox(label="Output")
         login_button = gr.Button("Login")
+        
+        def login(username, password):
+            message, token = login_user(username, password)
+            if token:
+                return f"{message}! Token: {token[:15]}... (truncated)"
+            return message
+            
         login_button.click(
-            login_interface,
+            login,
             inputs=[login_username, login_password],
             outputs=login_output
         )
 
-# You need to run Flask and Gradio separately
-if __name__ == "__main__":
-    # Run this in one terminal: python your_script.py --flask
-    # And in another: python your_script.py --gradio
-    import argparse
-    parser = argparse.ArgumentParser()
-    parser.add_argument('--flask', action='store_true')
-    parser.add_argument('--gradio', action='store_true')
-    args = parser.parse_args()
-    
-    if args.flask:
-        flask_app.run(port=5000)
-    elif args.gradio:
-        demo.launch(port=7860)
-    else:
-        print("Please specify --flask or --gradio")
+demo.launch()