Upload app.py with huggingface_hub

app.py

@@ -26,11 +26,16 @@ MODELS = {
     "8b": "unsloth/Llama-3.1-8B-Instruct",
 }
 
+# Duplicated here since spaces/app.py runs standalone on HF (can't import codewraith)
 SYSTEM_MESSAGE = (
     "You are CodeWraith, a technical specification generator. "
     "Given Python source code, produce a structured Markdown specification "
     "that accurately captures all functions, classes, parameters, return types, "
-    "dependencies, and error handling patterns."
+    "dependencies, and error handling patterns. "
+    "Include a mermaid diagram showing the relationships between classes and functions. "
+    "Use valid mermaid syntax with proper node IDs (no spaces or special characters in IDs). "
+    "Example: ```mermaid\ngraph TD\n    A[ModuleName] --> B[ClassName]\n"
+    "    B --> C[method_name]\n```"
 )
 
 EXAMPLE_CODE = '''\