Update Text to Image: Format dropdown, remove Aspect Ratio

- Replace Style dropdown with Format dropdown
- Options: image (generate_image_5), multi (generate_image_multi_5), 3d (generate_image_3d)
- Remove Aspect Ratio dropdown

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

app.py

@@ -23,13 +23,12 @@ def create_demo():
         theme=gr.themes.Soft()
     ) as demo:
         gr.Markdown("# StackNet 1:1 Preview Demo")
-        gr.Markdown("AI-powered media generation - Images, Videos, and Music")
 
         # API Key input at the top
         with gr.Accordion("Settings", open=False):
             api_key = gr.Textbox(
-                label="StackNet API Key",
-                placeholder="Enter your API key (e.g., gn_xxxxx)",
+                label="StackNet Key",
+                placeholder="Enter your key (e.g., sn_xxxxx)",
                 type="password"
             )
 
@@ -62,8 +61,7 @@ def create_demo():
             fn=Handlers.generate_image,
             inputs=[
                 tabs["text_to_image"]["prompt"],
-                tabs["text_to_image"]["style"],
-                tabs["text_to_image"]["aspect_ratio"],
+                tabs["text_to_image"]["format_type"],
                 api_key
             ],
             outputs=[

src/services/image.py

@@ -36,43 +36,33 @@ class ImageService:
     async def generate_image(
         self,
         prompt: str,
-        style: Optional[str] = None,
-        aspect_ratio: Optional[str] = None,
+        format_type: str = "image",
         on_progress: Optional[Callable[[float, str], None]] = None
     ) -> List[GeneratedImage]:
         """
-        Generate image from a text prompt using generate_image_5 tool.
+        Generate image from a text prompt.
 
         Args:
             prompt: Description of desired image
-            style: Style preset (Photorealistic, Digital Art, etc.)
-            aspect_ratio: Aspect ratio (1:1, 16:9, 9:16, etc.)
+            format_type: Format type - "image" (generate_image_5),
+                        "multi" (generate_image_multi_5), or "3d" (generate_image_3d)
             on_progress: Callback for progress updates
 
         Returns:
             List of generated images
         """
-        full_prompt = prompt
-        if style and style != "Photorealistic":
-            full_prompt = f"{prompt}, {style.lower()} style"
-
-        # Determine dimensions from aspect ratio
-        width, height = 1024, 1024
-        if aspect_ratio == "16:9":
-            width, height = 1280, 720
-        elif aspect_ratio == "9:16":
-            width, height = 720, 1280
-        elif aspect_ratio == "4:3":
-            width, height = 1024, 768
-        elif aspect_ratio == "3:4":
-            width, height = 768, 1024
+        # Select tool based on format type
+        if format_type == "multi":
+            tool_name = "generate_image_multi_5"
+        elif format_type == "3d":
+            tool_name = "generate_image_3d"
+        else:
+            tool_name = "generate_image_5"
 
         result = await self.client.submit_tool_task(
-            tool_name="generate_image_5",
+            tool_name=tool_name,
             parameters={
-                "prompt": full_prompt,
-                "width": width,
-                "height": height
+                "prompt": prompt
             },
             on_progress=on_progress
         )

src/ui/handlers.py

@@ -183,8 +183,7 @@ class Handlers:
     @staticmethod
     def generate_image(
         prompt: str,
-        style: str,
-        aspect_ratio: str,
+        format_type: str,
         api_key: str = ""
     ) -> Tuple[Optional[str], str]:
         """Handle text-to-image generation."""
@@ -204,8 +203,7 @@ class Handlers:
                 images = loop.run_until_complete(
                     service.generate_image(
                         prompt=prompt,
-                        style=style,
-                        aspect_ratio=aspect_ratio
+                        format_type=format_type
                     )
                 )
 

src/ui/tabs.py

@@ -156,19 +156,11 @@ def create_text_to_image_tab():
             lines=3
         )
 
-        with gr.Row():
-            style = gr.Dropdown(
-                label="Style",
-                choices=["Photorealistic", "Digital Art", "Oil Painting", "Watercolor", "Sketch", "Anime"],
-                value="Photorealistic",
-                scale=1
-            )
-            aspect_ratio = gr.Dropdown(
-                label="Aspect Ratio",
-                choices=["1:1", "16:9", "9:16", "4:3", "3:4"],
-                value="1:1",
-                scale=1
-            )
+        format_type = gr.Dropdown(
+            label="Format",
+            choices=["image", "multi", "3d"],
+            value="image"
+        )
 
         generate_btn = gr.Button("Generate Image", variant="primary", size="lg")
 
@@ -178,8 +170,7 @@ def create_text_to_image_tab():
 
     return {
         "prompt": prompt,
-        "style": style,
-        "aspect_ratio": aspect_ratio,
+        "format_type": format_type,
         "generate_btn": generate_btn,
         "status": status,
         "output_image": output_image