Update app.py

app.py

@@ -1,43 +1,31 @@
 import gradio as gr
 import spaces
-import torch
 from model import ModelHandler
 from generator import Generator
 from config import Config
 
-# 1. Initialize Models Globally
+# 1. Initialize Models
 print("Initializing Application...")
 handler = ModelHandler()
 handler.load_models()
 gen = Generator(handler)
 
-# 2. Define GPU-enabled Inference Function
+# 2. Define GPU Inference Function
 @spaces.GPU(duration=20)
-def process_img(
-    image, 
+def process_text(
     prompt,
     negative_prompt,
-    cfg_scale, # <-- RE-ENABLED
+    cfg_scale, 
     steps, 
-    img_strength, 
-    depth_strength, 
-    edge_strength,
     seed
 ):
-    if image is None:
-        raise gr.Error("Please upload an image first.")
-    
     try:
         print("--- Starting Generation ---")
         result = gen.predict(
-            image, 
-            prompt,
+            user_prompt=prompt,
             negative_prompt=negative_prompt,
-            guidance_scale=cfg_scale, # <-- RE-ENABLED
+            guidance_scale=cfg_scale,
             num_inference_steps=steps,
-            img2img_strength=img_strength,
-            depth_strength=depth_strength,
-            lineart_strength=edge_strength,
             seed=seed
         )
         print("--- Generation Complete ---")
@@ -48,30 +36,29 @@ def process_img(
         raise gr.Error(f"An error occurred: {str(e)}")
 
 # 3. Build Gradio Interface
-with gr.Blocks(title="Face To Style", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="Pixel Art Generator", theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
-        # 🎮 Face to Style
-        Upload any image. If there is a face, we'll keep the identity. If not, we'll stylize the scene!
+        # 🎮 Text to Pixel Art
+        Type a prompt to generate high-quality pixel art scenes.
         """
     )
     
     with gr.Row():
         with gr.Column(scale=2):
-            input_img = gr.Image(type="pil", label="Input Image")
             prompt = gr.Textbox(
-                label="Prompt (Optional)", 
-                placeholder="Leave empty for auto-captioning...",
-                info=f"The trigger words '{Config.STYLE_TRIGGER}' are added automatically."
+                label="Prompt", 
+                placeholder="e.g. cyberpunk city street at night, rain",
+                info="The trigger words 'p1x3l4rt, pixel art' are added automatically."
             )
             
             negative_prompt = gr.Textbox(
-                label="Negative Prompt (Optional)", 
+                label="Negative Prompt", 
                 placeholder="e.g., blurry, text, watermark, bad art...",
                 value=Config.DEFAULT_NEGATIVE_PROMPT
             )
             
-            with gr.Accordion("Advanced Settings", open=False):
+            with gr.Accordion("Settings", open=True):
                 seed = gr.Number(
                     label="Seed", 
                     value=-1, 
@@ -79,79 +66,48 @@ with gr.Blocks(title="Face To Style", theme=gr.themes.Soft()) as demo:
                     precision=0
                 )
                 
-                # --- RE-ENABLED CFG/GUIDANCE SLIDER ---
                 cfg_scale = gr.Slider(
                     elem_id="cfg_scale",
                     minimum=1.0, 
-                    maximum=10.0, # Range for TCD+Style
+                    maximum=5.0, 
                     step=0.1, 
-                    value=Config.CGF_SCALE, # Default 4.0
-                    label="Style Strength (Guidance)"
+                    value=Config.CGF_SCALE, 
+                    label="CFG Scale"
                 )
-                
                 steps = gr.Slider(
                     elem_id="steps",
-                    minimum=1, 
+                    minimum=4, 
                     maximum=20, 
                     step=1, 
-                    value=8, # TCD default
-                    label="Steps Number"
-                )
-                img_strength = gr.Slider(
-                    elem_id="img_strength",
-                    minimum=0.1, 
-                    maximum=1.0, 
-                    step=0.05, 
-                    value=Config.IMG_STRENGTH, 
-                    label="Image Strength (Img2Img)"
-                )
-                depth_strength = gr.Slider(
-                    elem_id="depth_strength",
-                    minimum=0.0, 
-                    maximum=1.0, 
-                    step=0.05, 
-                    value=Config.DEPTH_STRENGTH, 
-                    label="DepthMap Strength"
-                )
-                edge_strength = gr.Slider(
-                    elem_id="edge_strength",
-                    minimum=0.0, 
-                    maximum=1.0, 
-                    step=0.05, 
-                    value=Config.EDGE_STRENGTH, 
-                    label="EdgeMap Strength (LineArt)"
+                    value=Config.STEPS_NUMBER, 
+                    label="Steps"
                 )
             
             run_btn = gr.Button("Generate", variant="primary")
             
         with gr.Column(scale=1):
-            output_img = gr.Image(label="Styled Result")
+            output_img = gr.Image(label="Result")
                 
     # Event Handler
     all_inputs = [
-        input_img, 
         prompt, 
         negative_prompt,
-        cfg_scale, # <-- RE-ENABLED
+        cfg_scale, 
         steps, 
-        img_strength, 
-        depth_strength, 
-        edge_strength,
         seed
     ]
     
     run_btn.click(
-        fn=process_img, 
+        fn=process_text, 
         inputs=all_inputs, 
         outputs=[output_img]
     )
 
-
-# 4. Launch the App
+# 4. Launch
 if __name__ == "__main__":
     demo.queue(max_size=20, api_open=True)
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        show_api=True 
+        show_api=True
     )