Update app.py

app.py

@@ -1,91 +1,55 @@
-from flask import Flask, request, jsonify, make_response
+from flask import Flask, request, jsonify
 import time
-from datetime import datetime
 
 app = Flask(__name__)
 
-# Fake database simulation
+# Fake database
 fake_db = {
     "accounts": []
 }
 
-# Helper function to generate a new ID
 def generate_new_id():
     if not fake_db["accounts"]:
         return 1
     return max(account["id"] for account in fake_db["accounts"]) + 1
 
-def create_response(data=None, message="", status_code=200, error=None):
-    """Standard response format for all API endpoints"""
-    response = {
-        "meta": {
-            "timestamp": datetime.utcnow().isoformat(),
-            "status": status_code,
-            "message": message,
-            "error": error
-        },
-        "data": data
-    }
-    
-    headers = {
-        "Content-Type": "application/json",
-        "X-API-Version": "1.0",
-        "X-Server": "StableDiffusionAPI"
-    }
-    
-    return make_response(jsonify(response), status_code, headers)
-
-# Home page
 @app.route('/')
 def home():
-    return create_response(
-        data={"service": "stablediffusionapi test"},
-        message="Service is running",
-        status_code=200
-    )
+    return "stablediffusionapi test"
 
-# GET all accounts
 @app.route('/users', methods=['GET'])
 def get_all_accounts():
-    return create_response(
-        data={"users": fake_db["accounts"]},
-        message="Users retrieved successfully",
-        status_code=200
-    )
+    return jsonify({
+        "success": True,
+        "users": fake_db["accounts"],
+        "error": None
+    })
 
-# GET or CREATE account by platform and UDID
 @app.route('/account/<platform>/<udid>', methods=['GET'])
 def get_or_create_account(platform, udid):
-    # Validate platform (should be "IOS" or "AND")
     platform = platform.upper()
     if platform not in ["IOS", "AND"]:
-        return create_response(
-            data=None,
-            message="Invalid platform",
-            error="Platform must be either IOS or AND",
-            status_code=400
-        )
+        return jsonify({
+            "success": False,
+            "account": None,
+            "error": "Platform must be either IOS or AND"
+        }), 400
     
-    # Search for account in the database
     account = next((acc for acc in fake_db["accounts"] if acc["udid"] == udid), None)
     
     if account:
-        # Account found
-        return create_response(
-            data={
-                "account": {
-                    "id": account["id"],
-                    "timestamp": account["timestamp"],
-                    "udid": account["udid"],
-                    "platform": account["platform"],
-                    "credit": account["credit"]
-                }
+        return jsonify({
+            "success": True,
+            "account": {
+                "id": account["id"],
+                "timestamp": account["timestamp"],
+                "udid": account["udid"],
+                "platform": account["platform"],
+                "credit": account["credit"]
             },
-            message="Account retrieved successfully",
-            status_code=200
-        )
+            "error": None
+        })
     else:
-        # Account not found, create new account
         new_account = {
             "id": generate_new_id(),
             "timestamp": int(time.time()),
@@ -95,19 +59,17 @@ def get_or_create_account(platform, udid):
         }
         fake_db["accounts"].append(new_account)
         
-        return create_response(
-            data={
-                "account": {
-                    "id": new_account["id"],
-                    "timestamp": new_account["timestamp"],
-                    "udid": new_account["udid"],
-                    "platform": new_account["platform"],
-                    "credit": new_account["credit"]
-                }
+        return jsonify({
+            "success": True,
+            "account": {
+                "id": new_account["id"],
+                "timestamp": new_account["timestamp"],
+                "udid": new_account["udid"],
+                "platform": new_account["platform"],
+                "credit": new_account["credit"]
             },
-            message="Account created successfully",
-            status_code=201
-        )
+            "error": None
+        }), 201
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)