Update app.py

app.py

@@ -5,7 +5,35 @@ import time
 
 app = Flask(__name__)
 
-# get_token 和 req_flux 函数保持不变
+def get_token():
+    url = "https://fluxaiweb.com/flux/getToken"
+    response = requests.get(url)
+    if response.status_code == 200:
+        response_json = response.json()
+        return response_json.get("data", {}).get("token")
+    return None
+
+def req_flux(token, prompt_value, aspect_ratio="1:1", output_format="webp", num_outputs=1, output_quality=90):
+    url = "https://fluxaiweb.com/flux/generateImage"
+    payload = {
+        "prompt": prompt_value,
+        "aspectRatio": aspect_ratio,
+        "outputFormat": output_format,
+        "numOutputs": num_outputs,
+        "outputQuality": output_quality
+    }
+    headers = {
+        'Content-Type': 'application/json',
+        'token': token
+    }
+    try:
+        response = requests.post(url, headers=headers, json=payload)
+        response.raise_for_status()
+        data = response.json()
+        return data.get("data", {}).get("image")
+    except requests.exceptions.RequestException as e:
+        print(f"Error making request: {e}")
+        return None
 
 def generate_fake_stream(image_url):
     chunks = [