Update app.py

app.py

@@ -1,11 +1,12 @@
-from flask import Flask, request, jsonify
+from flask import Flask, request, jsonify, make_response
 import time
+from datetime import datetime
 
 app = Flask(__name__)
 
-# Fake database simulation - more generic name
+# Fake database simulation
 fake_db = {
-    "accounts": []  # Changed from "users" to "accounts" to better reflect the data
+    "accounts": []
 }
 
 # Helper function to generate a new ID
@@ -14,15 +15,43 @@ def generate_new_id():
         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 "stablediffusionapi test"
+    return create_response(
+        data={"service": "stablediffusionapi test"},
+        message="Service is running",
+        status_code=200
+    )
 
 # GET all accounts
 @app.route('/accounts', methods=['GET'])
 def get_all_accounts():
-    return jsonify({"accounts": fake_db["accounts"]})
+    return create_response(
+        data={"accounts": fake_db["accounts"]},
+        message="Accounts retrieved successfully",
+        status_code=200
+    )
 
 # GET or CREATE account by platform and UDID
 @app.route('/account/<platform>/<udid>', methods=['GET'])
@@ -30,43 +59,55 @@ def get_or_create_account(platform, udid):
     # Validate platform (should be "IOS" or "AND")
     platform = platform.upper()
     if platform not in ["IOS", "AND"]:
-        return jsonify({
-            "error": "Invalid platform. Use 'IOS' or 'AND'."
-        }), 400
+        return create_response(
+            data=None,
+            message="Invalid platform",
+            error="Platform must be either IOS or AND",
+            status_code=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 the account data
-        return jsonify({
-            "id": account["id"],
-            "timestamp": account["timestamp"],
-            "udid": account["udid"],
-            "platform": account["platform"],
-            "credit": account["credit"]
-        })
+        # Account found
+        return create_response(
+            data={
+                "account": {
+                    "id": account["id"],
+                    "timestamp": account["timestamp"],
+                    "udid": account["udid"],
+                    "platform": account["platform"],
+                    "credit": account["credit"]
+                }
+            },
+            message="Account retrieved successfully",
+            status_code=200
+        )
     else:
-        # Account not found, create new account with 100 credits
+        # Account not found, create new account
         new_account = {
             "id": generate_new_id(),
             "timestamp": int(time.time()),
             "udid": udid,
             "platform": platform,
-            "credit": 100  # Default credit
+            "credit": 100
         }
-        
-        # Add to database
         fake_db["accounts"].append(new_account)
         
-        # Return the newly created account
-        return jsonify({
-            "id": new_account["id"],
-            "timestamp": new_account["timestamp"],
-            "udid": new_account["udid"],
-            "platform": new_account["platform"],
-            "credit": new_account["credit"]
-        }), 201  # 201 Created status code
+        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"]
+                }
+            },
+            message="Account created successfully",
+            status_code=201
+        )
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)