Update app.py

app.py

@@ -32,7 +32,7 @@ DTYPE = (
 )
 
 MODEL_NAME = "Qwen/Qwen3.5-2B"
-CATEGORIES =["Query", "Caption", "Point", "Detect"]
+CATEGORIES = ["Query", "Caption", "Point", "Detect"]
 
 print(f"Loading model: {MODEL_NAME} on {DEVICE}...")
 try:
@@ -75,10 +75,10 @@ def generate_inference_stream(image: Image.Image, category: str, prompt: str):
     else:
         full_prompt = prompt
 
-    messages =[
+    messages = [
         {
             "role": "user",
-            "content":[
+            "content": [
                 {"type": "image", "image": image},
                 {"type": "text", "text": full_prompt},
             ],
@@ -208,7 +208,7 @@ async def homepage(request: Request):
 
         /* Nodes */
         .node {{
-            position: absolute; width: 320px;
+            position: absolute; width: 340px;
             background: var(--node-bg); border: 1px solid var(--node-border);
             border-radius: 8px; box-shadow: 0 8px 24px rgba(0,0,0,0.5);
             z-index: 10; display: flex; flex-direction: column;
@@ -244,7 +244,7 @@ async def homepage(request: Request):
         }}
         .file-upload:hover {{ border-color: var(--accent); }}
         
-        .img-preview {{ width: 100%; height: auto; max-height: 180px; object-fit: contain; border-radius: 4px; display: none; margin-top: 10px; background: #000; }}
+        .img-preview {{ width: 100%; height: auto; max-height: 280px; object-fit: contain; border-radius: 4px; display: none; margin-top: 10px; background: #000; }}
 
         select, textarea {{
             width: 100%; background: #121212; border: 1px solid var(--node-border);
@@ -254,11 +254,11 @@ async def homepage(request: Request):
         select:focus, textarea:focus {{ border-color: var(--accent); }}
         
         button {{
-            background: var(--accent); color: #fff; border: none;
+            background: #ff69b4; color: #fff; border: none;
             padding: 10px; border-radius: 6px; font-weight: 600; font-size: 14px;
             cursor: pointer; transition: background 0.2s; display: flex; justify-content: center; gap: 8px; align-items: center;
         }}
-        button:hover {{ background: var(--accent-hover); }}
+        button:hover {{ background: #ff1493; }}
         button:disabled {{ background: var(--node-border); cursor: not-allowed; color: #888; }}
 
         .output-box {{
@@ -286,7 +286,7 @@ async def homepage(request: Request):
 
         <!-- Node 1: Image Input -->
         <div class="node" id="node-img" style="left: 100px; top: 150px;">
-            <div class="node-header"><span>🖼️ Input Node</span><span>ID: 01</span></div>
+            <div class="node-header"><span>Input Node</span><span>ID: 01</span></div>
             <div class="node-body">
                 <div>
                     <label>Upload Image</label>
@@ -303,7 +303,7 @@ async def homepage(request: Request):
         <!-- Node 2: Task Execution -->
         <div class="node" id="node-task" style="left: 550px; top: 150px;">
             <div class="port in" id="port-task-in" style="top: 30%;"></div>
-            <div class="node-header"><span>⚙️ Task Node</span><span>ID: 02</span></div>
+            <div class="node-header"><span>Task Node</span><span>ID: 02</span></div>
             <div class="node-body">
                 <div>
                     <label>Task Category</label>
@@ -319,7 +319,7 @@ async def homepage(request: Request):
                     <textarea id="promptInput" rows="4" placeholder="e.g., Count the total number of boats and describe the environment."></textarea>
                 </div>
                 <button id="runBtn">
-                    <span>Queue Execution</span>
+                    <span>Quick Execution</span>
                     <span class="loader" id="btnLoader"></span>
                 </button>
             </div>
@@ -329,7 +329,7 @@ async def homepage(request: Request):
         <!-- Node 3: Output Display -->
         <div class="node" id="node-out" style="left: 1000px; top: 150px; width: 380px;">
             <div class="port in" id="port-out-in"></div>
-            <div class="node-header"><span>📝 Output Node</span><span>ID: 03</span></div>
+            <div class="node-header"><span>Output Node</span><span>ID: 03</span></div>
             <div class="node-body">
                 <label>Streamed Result</label>
                 <div class="output-box" id="outputBox">Results will stream here...</div>