Update app.py

app.py

@@ -87,15 +87,15 @@ def generate_code(prompt, language, model_name, temperature, max_tokens):
         return "⚠️ Please describe what you want to build."
     
     model = MODELS.get(model_name, "qwen2.5-coder:3b")
-    full_prompt = f"""Write {language} code for the following task:
-
-{prompt}
-
-Requirements:
-- Clean, production-ready code
-- Add helpful comments
-- Handle edge cases
-- Output ONLY the code in a markdown code block"""
+    full_prompt = (
+        f"Write {language} code for the following task:\n\n"
+        f"{prompt}\n\n"
+        "Requirements:\n"
+        "- Clean, production-ready code\n"
+        "- Add helpful comments\n"
+        "- Handle edge cases\n"
+        "- Output ONLY the code in a markdown code block"
+    )
 
     try:
         r = requests.post(
@@ -129,8 +129,7 @@ def explain_code(code, model_name, detail_level, max_tokens):
         "Detailed": "Give a detailed explanation including: purpose, how it works step-by-step, time/space complexity, and potential improvements."
     }
     
-    prompt = f"""{detail_prompts[detail_level]}
-{code}
+    prompt = detail_prompts[detail_level] + "\n\n" + code
 
     try:
         r = requests.post(
@@ -147,19 +146,16 @@ def fix_code(code, error, model_name, max_tokens):
         return "⚠️ Paste code to fix."
     
     model = MODELS.get(model_name, "qwen2.5-coder:3b")
-    prompt = f"""Fix the following code and explain what was wrong.
-
-Code:
-```
-{code}
-```
-
-Error/Problem: {error or 'Code is not working as expected'}
-
-Provide:
-1. The fixed code
-2. Brief explanation of what was wrong
-3. Any suggestions to prevent similar issues"""
+    error_msg = error if error else "Code is not working as expected"
+    prompt = (
+        "Fix the following code and explain what was wrong.\n\n"
+        "Code:\n" + code + "\n\n"
+        "Error/Problem: " + error_msg + "\n\n"
+        "Provide:\n"
+        "1. The fixed code\n"
+        "2. Brief explanation of what was wrong\n"
+        "3. Any suggestions to prevent similar issues"
+    )
 
     try:
         r = requests.post(
@@ -176,14 +172,15 @@ def review_code(code, model_name, max_tokens):
         return "⚠️ Paste code to review."
     
     model = MODELS.get(model_name, "qwen2.5-coder:3b")
-    prompt = f"""Review this code and provide feedback on:
-
-1. **Code Quality** - Is it clean, readable, well-structured?
-2. **Bugs/Issues** - Any potential bugs or problems?
-3. **Performance** - Any performance concerns?
-4. **Security** - Any security issues?
-5. **Suggestions** - How could it be improved?
-{code}
+    prompt = (
+        "Review this code and provide feedback on:\n\n"
+        "1. **Code Quality** - Is it clean, readable, well-structured?\n"
+        "2. **Bugs/Issues** - Any potential bugs or problems?\n"
+        "3. **Performance** - Any performance concerns?\n"
+        "4. **Security** - Any security issues?\n"
+        "5. **Suggestions** - How could it be improved?\n\n"
+        "Code:\n" + code
+    )
 
     try:
         r = requests.post(
@@ -202,10 +199,7 @@ footer { display: none !important; }
 
 with gr.Blocks(title="Axon v6", css=css, theme=gr.themes.Soft(primary_hue="violet", secondary_hue="blue")) as demo:
     
-    gr.Markdown("""
-    # 🔥 Axon v6
-    ### AI Coding Assistant • 10 Models • 100% Local • No Rate Limits
-    """)
+    gr.Markdown("# 🔥 Axon v6\n### AI Coding Assistant • 10 Models • 100% Local • No Rate Limits")
     
     status = gr.Markdown(value=get_status, every=5)