Update ui_manager.py

ui_manager.py

@@ -243,24 +243,24 @@ class UIManager:
     def create_interface(self):
         """Create professional user interface"""
 
-        css = CSSStyles.get_main_css()
-
-        # Gradio 4.x/5.x compatible initialization
-        blocks_kwargs = {
-            "title": "SceneWeaver - AI Background Generator",
-            "theme": gr.themes.Soft()
-        }
-
-        # Check Gradio version for css parameter compatibility
-        gradio_version = gr.__version__
-        major_version = int(gradio_version.split('.')[0])
-
-        if major_version >= 5:
-            # Gradio 5.x: use head parameter for CSS injection
-            blocks_kwargs["head"] = f"<style>{css}</style>"
+        self._css = CSSStyles.get_main_css()
+
+        # Check Gradio version for API compatibility
+        self._gradio_version = gr.__version__
+        self._gradio_major = int(self._gradio_version.split('.')[0])
+
+        # Gradio 5.x: css/theme moved to launch(), Blocks() has minimal params
+        # Gradio 4.x: css/theme are in Blocks() constructor
+        if self._gradio_major >= 5:
+            blocks_kwargs = {
+                "title": "SceneWeaver - AI Background Generator"
+            }
         else:
-            # Gradio 4.x: use css parameter directly
-            blocks_kwargs["css"] = css
+            blocks_kwargs = {
+                "css": self._css,
+                "title": "SceneWeaver - AI Background Generator",
+                "theme": gr.themes.Soft()
+            }
 
         with gr.Blocks(**blocks_kwargs) as interface:
 
@@ -515,12 +515,18 @@ class UIManager:
         """Launch the UI interface"""
         interface = self.create_interface()
 
-        return interface.launch(
-            share=share,
-            debug=debug,
-            show_error=True,
-            height=800,
-            favicon_path=None,
-            ssl_verify=False,
-            quiet=False
-        )
+        # Build launch kwargs based on Gradio version
+        launch_kwargs = {
+            "share": share,
+            "debug": debug,
+            "show_error": True,
+            "quiet": False
+        }
+
+        # Gradio 5.x: css/theme are passed to launch()
+        # Gradio 4.x: these were already set in Blocks()
+        if self._gradio_major >= 5:
+            launch_kwargs["css"] = self._css
+            launch_kwargs["theme"] = gr.themes.Soft()
+
+        return interface.launch(**launch_kwargs)