Update app.py

app.py

@@ -47,14 +47,18 @@ def process_and_package(ai_response: str):
     agent = LlamaCppAgent(
         provider,
         system_prompt=(
-            "You are a code parsing assistant. The user will paste AI output referencing multiple code files. "
-            "Return a JSON object mapping each filename to its full source code text. "
-            "Do not return anything else."
+            "You are a code packager. The user will paste text that may contain multiple code blocks. "
+            "Extract all code blocks, associate each with a filename if available, and return a JSON object "
+            "where keys are filenames and values are the full content. "
+            "ONLY return a raw JSON object — no explanations, no markdown, no text outside the JSON. "
+            "Example:\n\n"
+            "{\n  \"app.py\": \"import gradio as gr\\n...\",\n  \"utils.py\": \"def helper():\\n...\"\n}\n"
         ),
         predefined_messages_formatter_type=MessagesFormatterType.GEMMA_2,
         debug_output=True
     )
 
+
     # Call the agent
     # Use get_chat_response (not run)
     try: