flux2api

Paused

App Files Files Community

smgc commited on Aug 28, 2024

Commit

e1b2102

verified ·

1 Parent(s): 61edc08

Update app.py

Browse files

Files changed (1) hide show

app.py +48 -25

app.py CHANGED Viewed

@@ -6,22 +6,24 @@ import random
 import logging
 import sys
 import re
-from logging.handlers import RotatingFileHandler
 app = Flask(__name__)
 # 配置日志
 class RequestFormatter(logging.Formatter):
     def format(self, record):
-        record.url = request.url
-        record.remote_addr = request.remote_addr
-        return super().format(record)
 formatter = RequestFormatter(
-    '%(asctime)s - %(remote_addr)s - %(url)s - %(levelname)s - %(message)s'
 )
-handler = RotatingFileHandler('app.log', maxBytes=10000, backupCount=3)
 handler.setFormatter(formatter)
 handler.setLevel(logging.INFO)
@@ -110,27 +112,48 @@ def extract_params_from_prompt(prompt):
 @app.route('/')
 def index():
     usage = """
-    Welcome to the text-to-image API with siliconflow!
-    Usage:
-    1. Send a POST request to /ai/v1/chat/completions
-    2. Include your prompt in the 'content' field of the last message
-    3. Optional parameters:
-       -s <ratio>: Set image size ratio (e.g., -s 1:1, -s 16:9)
-       -o: Use original prompt without enhancement
-    Example:
-    {
-        "model": "flux",
-        "messages": [
-            {
-                "role": "user",
-                "content": "A beautiful landscape -s 16:9"
-            }
-        ]
-    }
-    For more details, please refer to the API documentation.
     """
     return usage, 200

 import logging
 import sys
 import re
+from logging.handlers import TimedRotatingFileHandler
 app = Flask(__name__)
 # 配置日志
 class RequestFormatter(logging.Formatter):
     def format(self, record):
+        if request.method == 'POST':
+            record.url = request.url
+            record.remote_addr = request.remote_addr
+            return super().format(record)
+        return None
 formatter = RequestFormatter(
+    '%(asctime)s - %(remote_addr)s - %(url)s - %(message)s'
 )
+handler = TimedRotatingFileHandler('app.log', when="midnight", interval=1, backupCount=30)
 handler.setFormatter(formatter)
 handler.setLevel(logging.INFO)
 @app.route('/')
 def index():
     usage = """
+    <html>
+    <head>
+        <title>Text-to-Image API with SiliconFlow</title>
+        <style>
+            body { font-family: Arial, sans-serif; line-height: 1.6; padding: 20px; max-width: 800px; margin: 0 auto; }
+            h1 { color: #333; }
+            h2 { color: #666; }
+            pre { background-color: #f4f4f4; padding: 10px; border-radius: 5px; }
+            code { font-family: Consolas, monospace; }
+        </style>
+    </head>
+    <body>
+        <h1>Welcome to the Text-to-Image API with SiliconFlow!</h1>
+        <h2>Usage:</h2>
+        <ol>
+            <li>Send a POST request to <code>/ai/v1/chat/completions</code></li>
+            <li>Include your prompt in the 'content' field of the last message</li>
+            <li>Optional parameters:
+                <ul>
+                    <li><code>-s &lt;ratio&gt;</code>: Set image size ratio (e.g., -s 1:1, -s 16:9)</li>
+                    <li><code>-o</code>: Use original prompt without enhancement</li>
+                </ul>
+            </li>
+        </ol>
+        <h2>Example Request:</h2>
+        <pre><code>
+{
+    "model": "flux",
+    "messages": [
+        {
+            "role": "user",
+            "content": "A beautiful landscape -s 16:9"
+        }
+    ]
+}
+        </code></pre>
+        <p>For more details, please refer to the API documentation.</p>
+    </body>
+    </html>
     """
     return usage, 200