Fix: Resolve Gradio 6.0 compatibility and gr.Div attribute error

app.py

@@ -2,10 +2,8 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 from PIL import Image
-import io
 
 # Initialize the HF Inference Client
-# It will use the HF_TOKEN from your Space's environment variables
 client = InferenceClient(token=os.getenv("HF_TOKEN"))
 
 MODELS = {
@@ -19,30 +17,24 @@ def generate_image(prompt, model_choice):
         
     model_id = MODELS.get(model_choice, MODELS["Flux.1 [Schnell]"])
     
-    print(f"[Vision] Request: {prompt[:50]}... via {model_choice}")
-    
     try:
-        # Use the inference client to get an image
-        # This returns a PIL.Image object directly
         image = client.text_to_image(prompt, model=model_id)
         return image
     except Exception as e:
-        print(f"[Error] Generation failed: {str(e)}")
-        # Raise error to show in Gradio UI
         raise gr.Error(f"Neural Core Error: {str(e)}")
 
 # UI Styling
 custom_css = """
 footer {visibility: hidden}
-.gradio-container { background-color: #030303 !important; }
-#title-container { text-align: center; margin-bottom: 20px; }
+body { background-color: #030303 !important; }
+.gradio-container { border: none !important; }
 #gen-button { background: #ffffff !important; color: #000000 !important; font-weight: 900 !important; }
 """
 
-with gr.Blocks(css=custom_css, theme=gr.themes.Default(primary_hue="zinc", secondary_hue="zinc")) as demo:
-    with gr.Div(elem_id="title-container"):
-        gr.Markdown("# CODEX_VISION")
-        gr.Markdown("NEURAL PROJECTION INTERFACE")
+# Fixed: Removed gr.Div (unsupported) and moved theme/css to launch() for Gradio 6 compatibility
+with gr.Blocks() as demo:
+    gr.Markdown("# CODEX_VISION", elem_id="title")
+    gr.Markdown("### NEURAL PROJECTION INTERFACE")
     
     with gr.Row():
         with gr.Column(scale=1):
@@ -68,4 +60,8 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Default(primary_hue="zinc", secon
     )
 
 if __name__ == "__main__":
-    demo.queue().launch()
+    # Gradio 6.0 compatible launch
+    demo.queue().launch(
+        css=custom_css, 
+        theme=gr.themes.Default(primary_hue="zinc", secondary_hue="zinc")
+    )