Update app.py

app.py

@@ -35,22 +35,8 @@ severities = {
     "Best Practice": "Low"
 }
 
-# ---------- Mock Knowledge Base ----------
-salesforce_knowledge_base = {
-    "governor limits soql": "In Salesforce, the governor limit for SOQL queries is 100 per synchronous transaction and 200 per asynchronous transaction.",
-    "governor limits dml": "The governor limit for DML statements is 150 per transaction.",
-    "bulkify apex trigger": "...",
-    "soql injection": "...",
-    "lwc best practices": "...",
-    "batch apex": "..."
-}
-
-# ---------- Load QnA Model ----------
-try:
-    qa_pipeline = pipeline("text2text-generation", model="google/flan-t5-large")
-except Exception as e:
-    print(f"Model loading error: {e}. Falling back to flan-t5-base.")
-    qa_pipeline = pipeline("text2text-generation", model="google/flan-t5-base")
+# ---------- Load QnA Model (no fallback) ----------
+qa_pipeline = pipeline("text2text-generation", model="google/flan-t5-large")
 
 # ---------- Logging ----------
 def log_to_console(data, log_type):
@@ -154,7 +140,7 @@ def validate_metadata(metadata, admin_id=None):
 
     return mtype, issue, recommendation
 
-# ---------- Salesforce Chatbot (Updated Prompt) ----------
+# ---------- Salesforce Chatbot (Improved Prompt) ----------
 conversation_history = []
 
 def salesforce_chatbot(query, history=[]):
@@ -168,28 +154,20 @@ def salesforce_chatbot(query, history=[]):
     ]
 
     if not any(keyword.lower() in query.lower() for keyword in salesforce_keywords):
-        return "Please ask a Salesforce-related question (e.g., Apex, SOQL, LWC, limits, etc)."
-
-    query_key = query.lower().strip()
-    for kb_key, kb_answer in salesforce_knowledge_base.items():
-        if kb_key in query_key:
-            conversation_history.append((query, kb_answer))
-            conversation_history = conversation_history[-6:]
-            log_to_console({"Question": query, "Answer": kb_answer}, "Chatbot Query")
-            return kb_answer
+        return "Please ask a Salesforce-related question."
 
     history_summary = "\n".join([f"User: {q}\nAssistant: {a}" for q, a in conversation_history[-4:]])
 
     prompt = f"""
-You are an expert Salesforce developer and certified architect. You provide 100% accurate, clear, and practical answers for topics like Apex, SOQL, LWC, governor limits, triggers, metadata, and more.
+You are a certified Salesforce developer and architect. Your role is to answer with 100% accurate and detailed technical explanations, especially about limits, code, and platform best practices.
 
-When answering:
-- ALWAYS give at least 2 lines of explanation.
-- Be clear, concise, and technically correct.
-- Mention official Salesforce limits if applicable.
-- Use bullet points or code snippets when helpful.
-- Avoid speculation — if unknown, say so and suggest Trailhead or official docs.
-- Examples must be realistic and follow best practices.
+Your answers MUST:
+- Always be at least two lines long.
+- Be correct, clear, and production-safe.
+- Include official Salesforce governor limits when applicable.
+- Use bullet points or code snippets when needed.
+- Recommend Trailhead or official docs if the answer isn't definitive.
+- Follow real-world practices (bulkification, error handling, etc).
 
 Conversation History:
 {history_summary}
@@ -204,8 +182,8 @@ Assistant:
         if output.startswith("Assistant:"):
             output = output.replace("Assistant:", "").strip()
 
-        if len(output) < 20:
-            output = f"I'm sorry, I couldn't generate a detailed answer for '{query}'. You can also check https://developer.salesforce.com/docs."
+        if len(output.split()) < 15:
+            output += "\n\nRefer to: https://developer.salesforce.com/docs for more."
 
         conversation_history.append((query, output))
         conversation_history = conversation_history[-6:]
@@ -236,7 +214,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     with gr.Tab("Salesforce Chatbot"):
         chatbot_output = gr.Chatbot(label="Conversation History", height=400)
-        query_input = gr.Textbox(label="Your Question", placeholder="e.g., What is heap size in Apex?")
+        query_input = gr.Textbox(label="Your Question", placeholder="e.g., How many DML operations are allowed in Apex?")
         with gr.Row():
             chatbot_button = gr.Button("Ask")
             clear_button = gr.Button("Clear Chat")