Update app.py

app.py

@@ -13,6 +13,7 @@ import sqlite3
 import hashlib
 import secrets
 import random
+import io
 from datetime import datetime, timedelta
 from pathlib import Path
 from typing import Optional, List, Dict, Any
@@ -32,7 +33,6 @@ except ImportError:
 # image processing
 try:
     from PIL import Image
-    import io
     IMAGE_OK = True
 except ImportError:
     IMAGE_OK = False
@@ -745,7 +745,7 @@ def get_frontend_html() -> str:
 
 <!-- Hidden Gradio Bridge Elements -->
 <div id="gradio-bridge" class="hidden-gradio">
-    {gradio_bridge_html}
+    __GRADIO_BRIDGE_HTML__
 </div>
 
 <script>
@@ -961,8 +961,11 @@ def get_gradio_bridge_html() -> str:
 def build_app():
     init_db()
     
-    with gr.Blocks(css=CUSTOM_CSS, title="OmniParse AI") as app:
-        gr.HTML(get_frontend_html().format(gradio_bridge_html=get_gradio_bridge_html()))
+    # In Gradio 6.0, CSS is passed to launch()
+    with gr.Blocks(title="OmniParse AI") as app:
+        # Safely concatenate HTML to avoid str.format() errors with CSS braces
+        final_html = get_frontend_html().replace("__GRADIO_BRIDGE_HTML__", get_gradio_bridge_html())
+        gr.HTML(final_html)
         
         # Hidden Gradio Elements for Bridge
         with gr.Row(visible=False):
@@ -1012,4 +1015,4 @@ def build_app():
 
 if __name__ == "__main__":
     app = build_app()
-    app.launch(server_name="0.0.0.0", server_port=7860, show_error=True)
+    app.launch(server_name="0.0.0.0", server_port=7860, show_error=True, css=CUSTOM_CSS)