Upload app.py with huggingface_hub

app.py

@@ -205,8 +205,27 @@ def generate_spec(
 
 
 def create_app():
+    mermaid_css = """
+    .mermaid .node rect,
+    .mermaid .node polygon,
+    .mermaid .node circle {
+        fill: #e8f0fe !important;
+        stroke: #4a6fa5 !important;
+    }
+    .mermaid .nodeLabel,
+    .mermaid .edgeLabel,
+    .mermaid text {
+        color: #1a1a1a !important;
+        fill: #1a1a1a !important;
+    }
+    .mermaid .edgePath .path {
+        stroke: #4a6fa5 !important;
+    }
+    """
+
     with gr.Blocks(
         title="CodeWraith - Module-to-Spec Transformer",
+        css=mermaid_css,
     ) as app:
         gr.Markdown(
             "# CodeWraith\n"
@@ -224,7 +243,7 @@ def create_app():
         with gr.Row():
             temperature = gr.Slider(0.0, 2.0, value=0.7, step=0.1, label="Temperature")
             top_p = gr.Slider(0.0, 1.0, value=0.9, step=0.05, label="Top-p")
-            max_tokens = gr.Slider(256, 8192, value=2048, step=256, label="Max Tokens")
+            max_tokens = gr.Slider(256, 8192, value=4096, step=256, label="Max Tokens")
         with gr.Row():
             use_rag = gr.Checkbox(value=True, label="Use RAG (retrieve similar examples)")
             generate_btn = gr.Button("Generate Specification", variant="primary")