Update app.py

app.py

@@ -232,6 +232,56 @@ def analyze_business_pdf(pdf_file):
 
     return app_title, screens
 
+# -----------------------
+# Dynamic Top Bar Generator
+# -----------------------
+def generate_topbar(screen_name):
+    """Generate a dynamic top bar layout depending on screen name."""
+    screen_name_lower = screen_name.lower()
+
+    # Common SVGs for top bar icons
+    ICONS = {
+        "back": """<svg viewBox="0 0 24 24" width="18" height="18" fill="#fff"><path d="M15.41 7.41 14 6l-6 6 6 6 1.41-1.41L10.83 12z"/></svg>""",
+        "menu": """<svg viewBox="0 0 24 24" width="18" height="18" fill="#fff"><path d="M3 6h18v2H3zM3 12h18v2H3zM3 18h18v2H3z"/></svg>""",
+        "share": """<svg viewBox="0 0 24 24" width="18" height="18" fill="#fff"><path d="M18 16.08c-.76 0-1.44.3-1.96.77L8.91 12.7a2.5 2.5 0 000-1.4l7.02-4.11A2.5 2.5 0 0018 7.91a2.5 2.5 0 10-2.5-2.5 2.5 2.5 0 00.1.71L8.59 10.3a2.5 2.5 0 100 3.4l7.02 4.11a2.5 2.5 0 00-.1.71 2.5 2.5 0 102.5-2.44z"/></svg>""",
+        "add": """<svg viewBox="0 0 24 24" width="18" height="18" fill="#fff"><path d="M19 11h-6V5h-2v6H5v2h6v6h2v-6h6z"/></svg>""",
+        "check": """<svg viewBox="0 0 24 24" width="18" height="18" fill="#fff"><path d="M9 16.17 4.83 12l-1.42 1.41L9 19l12-12-1.41-1.41z"/></svg>""",
+        "info": """<svg viewBox="0 0 24 24" width="18" height="18" fill="#fff"><path d="M11 9h2V7h-2v2zm1-7C6.48 2 2 6.48 2 12s4.48 10 10 10 10-4.48 10-10S17.52 2 12 2zm1 15h-2v-6h2v6z"/></svg>""",
+        "filter": """<svg viewBox="0 0 24 24" width="18" height="18" fill="#fff"><path d="M10 18h4v-2h-4v2zm-7-7v2h18v-2H3zm3-5v2h12V6H6z"/></svg>"""
+    }
+
+    # Default top bar (Dashboard-style)
+    left_icon = ICONS["menu"]
+    right_icons = ICONS["share"] + ICONS["info"]
+
+    # Adjust icons dynamically
+    if "dashboard" in screen_name_lower:
+        left_icon = ICONS["menu"]
+        right_icons = ICONS["share"] + ICONS["info"]
+
+    elif "repository" in screen_name_lower or "library" in screen_name_lower:
+        left_icon = ICONS["menu"]
+        right_icons = ICONS["filter"] + ICONS["add"]
+
+    elif "task" in screen_name_lower:
+        left_icon = ICONS["menu"]
+        right_icons = ICONS["filter"]
+
+    elif "form" in screen_name_lower or "edit" in screen_name_lower or "add" in screen_name_lower:
+        left_icon = ICONS["back"]
+        right_icons = ICONS["check"]
+
+    else:
+        left_icon = ICONS["menu"]
+        right_icons = ICONS["share"]
+
+    return f"""
+    <div class="topbar">
+        <div style='display:flex;align-items:center;gap:8px;'>{left_icon}<span>{screen_name}</span></div>
+        <div style='margin-left:auto;display:flex;align-items:center;gap:10px;'>{right_icons}</div>
+    </div>
+    """
+
 # -----------------------
 # Generate HTML + Image per Screen
 # -----------------------
@@ -241,6 +291,7 @@ def generate_mockups(app_title, screens):
         label = screen.get("screen_name", "Screen")
         html_content = screen.get("html", "<h2>Empty Screen</h2>")
         sidebar_html = generate_sidebar(app_title, screens, active_label=label)
+        topbar_html = generate_topbar(label)
         full_html = BASE_TEMPLATE.replace("{user_sidebar}", sidebar_html)\
                                  .replace("{user_content}", html_content)\
                                  .replace("{app_title}", app_title)