Update app.py

app.py

@@ -26,6 +26,7 @@ suggestions = {
     "Security": "Avoid dynamic SOQL. Use binding variables.",
     "Best Practice": "Refactor for readability and use bulk-safe patterns."
 }
+
 severities = {
     "Performance": "Medium",
     "Error": "High",
@@ -33,7 +34,7 @@ severities = {
     "Best Practice": "Low"
 }
 
-# ---------- Load Dummy Classifier & QnA Model ----------
+# ---------- Load QnA Model ----------
 qa_pipeline = pipeline("text2text-generation", model="google/flan-t5-base")
 
 # ---------- Code Analyzer ----------
@@ -82,22 +83,41 @@ def validate_metadata(metadata):
 
     return mtype, issue, recommendation
 
-# ---------- Natural Language Assistant ----------
+# ---------- Natural Language Assistant (Improved Prompt) ----------
 def process_nlp_query(query):
     if not query.strip():
         return "No question provided."
 
-    prompt = (
-        "You are a Salesforce Apex expert. "
-        "Answer the question clearly and include governor limits, code best practices, and real examples.\n\n"
-        f"Question: {query.strip()}\n\nAnswer:"
-    )
+    # Few-shot examples included in the prompt to simulate training
+    prompt = f"""
+You are an expert Salesforce Apex developer and trainer. Your job is to answer questions accurately, concisely, and based strictly on Salesforce best practices and official documentation. Always include real values like governor limits, correct terminology, and coding insights.
+
+Examples:
+
+Q: How many DML operations are allowed in Apex?
+A: You can perform up to 150 DML statements per Apex transaction.
+
+Q: How many SOQL queries are allowed in Apex?
+A: You can perform up to 100 SOQL queries per Apex transaction.
+
+Q: What is a trigger in Salesforce?
+A: A trigger is a piece of Apex code that executes before or after records are inserted, updated, deleted, or undeleted. Triggers help automate tasks and enforce business rules.
+
+Q: What are governor limits in Apex?
+A: Governor limits are runtime limits enforced by Salesforce to ensure efficient multi-tenancy. They include limits like 100 SOQL queries, 150 DML statements, and 10,000 records returned in queries per transaction.
+
+Q: What is bulkification in Apex?
+A: Bulkification is the practice of writing code that can handle multiple records at once, such as using loops and collections with DML to avoid hitting governor limits.
+
+Now answer this new question as an expert would:
+
+Q: {query.strip()}
+A:
+"""
 
     try:
         result = qa_pipeline(prompt, max_new_tokens=256, do_sample=False)
         output = result[0]["generated_text"]
-        if "Answer:" in output:
-            return output.split("Answer:")[-1].strip()
         return output.strip()
     except Exception as e:
         return f"⚠️ AI Response Error: {str(e)}"
@@ -130,4 +150,4 @@ with gr.Blocks() as demo:
 
 # ---------- Start UI ----------
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()