Update app.py

app.py

@@ -5,42 +5,14 @@ import time
 
 app = Flask(__name__)
 
-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
+# get_token 和 req_flux 函数保持不变
 
 def generate_fake_stream(image_url):
     chunks = [
         "I'm generating an image based on your prompt...",
         "The image is being processed...",
         "Almost there...",
-        f"Image generated successfully! You can view it here: {image_url}"
+        f"Image generated successfully! Here it is:\n\n![Generated Image]({image_url})"
     ]
     
     for i, chunk in enumerate(chunks):
@@ -82,7 +54,7 @@ def chat_completions():
                     "index": 0,
                     "message": {
                         "role": "assistant",
-                        "content": f"I've generated an image based on your prompt. You can view it here: {image_url}"
+                        "content": f"I've generated an image based on your prompt. Here it is:\n\n![Generated Image]({image_url})"
                     },
                     "finish_reason": "stop"
                 }