Update app.py

app.py

@@ -34,11 +34,11 @@ severities = {
     "Best Practice": "Low"
 }
 
-# ---------- Mock Salesforce Knowledge Base ----------
+# ---------- Mock Knowledge Base ----------
 salesforce_knowledge_base = {
-    "governor limits soql": "...",  # Truncated for brevity
-    "governor limits dml": "...",
-    "bulkify apex trigger": "...",
+    "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": "...",  # Truncated for brevity
     "soql injection": "...",
     "lwc best practices": "...",
     "batch apex": "..."
@@ -51,27 +51,56 @@ 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")
 
-# ---------- Local Logging ----------
+# ---------- Logging ----------
 def log_to_console(data, log_type):
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     print(f"[{timestamp}] {log_type} Log: {data}")
 
-# ---------- Code Analyzer ----------
+# ---------- Salesforce Connection ----------
+try:
+    session_id, instance = SalesforceLogin(
+        username=SF_USERNAME,
+        password=SF_PASSWORD,
+        security_token=SF_SECURITY_TOKEN
+    )
+    sf = Salesforce(instance=instance, session_id=session_id)
+    print("✅ Connected to Salesforce successfully")
+except Exception as e:
+    sf = None
+    print(f"❌ Failed to connect to Salesforce: {e}")
+
+# ---------- Code Analyzer (Salesforce Integration) ----------
 def analyze_code(code):
     if not code.strip():
         return "No code provided.", "", ""
+
     label = random.choice(list(label_to_issue_type.keys()))
     issue_type = label_to_issue_type[label]
     suggestion = suggestions[issue_type]
     severity = severities[issue_type]
-    log_data = {
+
+    review_data = {
         "Name": f"Review_{issue_type}",
-        "CodeSnippet": code,
-        "IssueType": issue_type,
-        "Suggestion": suggestion,
-        "Severity": severity
+        "CodeSnippet__c": code,
+        "IssueType__c": issue_type,
+        "Suggestion__c": suggestion,
+        "Severity__c": severity
     }
-    log_to_console(log_data, "Code Review")
+
+    log_to_console(review_data, "Code Review")
+
+    if sf:
+        try:
+            result = sf.CodeReviewResult__c.create(review_data)
+            if result.get("success"):
+                log_to_console({"Salesforce Record ID": result["id"]}, "Salesforce Create")
+            else:
+                log_to_console(result, "Salesforce Error")
+        except Exception as e:
+            log_to_console({"Salesforce Exception": str(e)}, "Salesforce Error")
+    else:
+        log_to_console("Salesforce not connected.", "Salesforce Error")
+
     return issue_type, suggestion, severity
 
 # ---------- Metadata Validator ----------
@@ -93,13 +122,20 @@ def validate_metadata(metadata):
 
 # ---------- Salesforce Chatbot ----------
 conversation_history = []
+
 def salesforce_chatbot(query, history=[]):
     global conversation_history
     if not query.strip():
         return "Please provide a valid Salesforce-related question."
-    salesforce_keywords = [ "apex", "soql", "trigger", "lwc", "governor limits", "dml", "metadata", "batch apex", "api", "profile", "workflow", "custom object", "record type", "sfdc", "force.com" ]
-    if not any(keyword in query.lower() for keyword in salesforce_keywords):
-        return "Please ask a Salesforce-related question (e.g., about Apex, SOQL, LWC, or Salesforce platform features)."
+
+    salesforce_keywords = [
+        "apex", "soql", "trigger", "lwc", "aura", "visualforce", "salesforce", "governor limits",
+        "dml", "metadata", "batch apex", "queueable", "future method", "api", "sfdc"
+    ]
+
+    if not any(keyword.lower() in query.lower() for keyword in salesforce_keywords):
+        return "Please ask a Salesforce-related question."
+
     query_key = query.lower().strip()
     for kb_key, kb_answer in salesforce_knowledge_base.items():
         if kb_key in query_key:
@@ -107,21 +143,25 @@ def salesforce_chatbot(query, history=[]):
             conversation_history = conversation_history[-6:]
             log_to_console({"Question": query, "Answer": kb_answer}, "Chatbot Query")
             return kb_answer
+
     history_summary = "\n".join([f"User: {q}\nAssistant: {a}" for q, a in conversation_history[-4:]])
     prompt = f"""
 You are an expert Salesforce developer...
+
 Conversation History:
 {history_summary}
+
 Question: {query.strip()}
 Answer:
 """
+
     try:
         result = qa_pipeline(prompt, max_new_tokens=1024, do_sample=False, temperature=0.1, top_k=50)
         output = result[0]["generated_text"].strip()
         if output.startswith("Answer:"):
             output = output[7:].strip()
         if len(output) < 20:
-            output = f"I'm sorry, I couldn't find a precise answer for '{query}'. Please check Salesforce docs."
+            output = f"I'm sorry, I couldn't find a precise answer for '{query}'. Please refer to Salesforce docs."
         conversation_history.append((query, output))
         conversation_history = conversation_history[-6:]
         log_to_console({"Question": query, "Answer": output}, "Chatbot Query")
@@ -129,20 +169,7 @@ Answer:
     except Exception as e:
         return f"⚠️ Error generating response: {str(e)}"
 
-# ---------- Salesforce Connection ----------
-try:
-    session_id, instance = SalesforceLogin(
-        username=SF_USERNAME,
-        password=SF_PASSWORD,
-        security_token=SF_SECURITY_TOKEN
-    )
-    sf = Salesforce(instance=instance, session_id=session_id)
-    print("✅ Connected to Salesforce successfully")
-except Exception as e:
-    sf = None
-    print(f"❌ Failed to connect to Salesforce: {e}")
-
-# ---------- Create Salesforce Record ----------
+# ---------- Create Salesforce Record (Generic Manual Entry) ----------
 def create_salesforce_record(object_type, field_name, field_value):
     if not sf:
         return "Salesforce connection not established."
@@ -161,7 +188,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🤖 Advanced Salesforce AI Code Review & Chatbot")
 
     with gr.Tab("Code Review"):
-        code_input = gr.Textbox(label="Apex / LWC Code", lines=8, placeholder="Enter code here")
+        code_input = gr.Textbox(label="Apex / LWC Code", lines=8, placeholder="Enter your Apex or LWC code here")
         issue_type = gr.Textbox(label="Issue Type")
         suggestion = gr.Textbox(label="AI Suggestion")
         severity = gr.Textbox(label="Severity")
@@ -169,7 +196,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         code_button.click(analyze_code, inputs=code_input, outputs=[issue_type, suggestion, severity])
 
     with gr.Tab("Metadata Validation"):
-        metadata_input = gr.Textbox(label="Metadata XML", lines=8)
+        metadata_input = gr.Textbox(label="Metadata XML", lines=8, placeholder="Enter your metadata XML here")
         mtype = gr.Textbox(label="Type")
         issue = gr.Textbox(label="Issue")
         recommendation = gr.Textbox(label="Recommendation")
@@ -178,7 +205,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")
+        query_input = gr.Textbox(label="Your Question", placeholder="e.g., How do I bulkify an Apex trigger?")
         with gr.Row():
             chatbot_button = gr.Button("Ask")
             clear_button = gr.Button("Clear Chat")
@@ -186,7 +213,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
         def update_chatbot(query, chat_history):
             if not query.strip():
-                return chat_history, "Please enter a question."
+                return chat_history, "Please enter a valid question."
             response = salesforce_chatbot(query, chat_history)
             chat_history.append((query, response))
             return chat_history, ""
@@ -200,7 +227,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         clear_button.click(fn=clear_chat, inputs=None, outputs=[chatbot_output, query_input])
 
     with gr.Tab("Create Salesforce Record"):
-        gr.Markdown("### 📇 Create a New Salesforce Record")
+        gr.Markdown("### 📇 Manually Create Salesforce Record")
         object_type = gr.Textbox(label="Object API Name", placeholder="e.g., Account")
         field_name = gr.Textbox(label="Field API Name", placeholder="e.g., Name")
         field_value = gr.Textbox(label="Field Value", placeholder="e.g., Test Company")