Update app.py

app.py

@@ -3,7 +3,7 @@ import random
 import gradio as gr
 from datetime import datetime
 from transformers import pipeline
-from simple_salesforce import Salesforce, SalesforceLogin, SFType
+from simple_salesforce import Salesforce, SalesforceLogin
 from dotenv import load_dotenv
 
 # ---------- Load Environment Variables ----------
@@ -69,7 +69,7 @@ except Exception as e:
     sf = None
     print(f"❌ Failed to connect to Salesforce: {e}")
 
-# ---------- Code Analyzer (Salesforce Integration) ----------
+# ---------- Code Analyzer (with Salesforce create) ----------
 def analyze_code(code):
     if not code.strip():
         return "No code provided.", "", ""
@@ -169,20 +169,6 @@ Answer:
     except Exception as e:
         return f"⚠️ Error generating response: {str(e)}"
 
-# ---------- Create Salesforce Record (Generic Manual Entry) ----------
-def create_salesforce_record(object_type, field_name, field_value):
-    if not sf:
-        return "Salesforce connection not established."
-    try:
-        sobject = SFType(object_type, sf.session_id, sf.sf_instance)
-        result = sobject.create({field_name: field_value})
-        if result.get("success"):
-            return f"✅ Record created in {object_type} with ID: {result['id']}"
-        else:
-            return f"❌ Failed to create record: {result}"
-    except Exception as e:
-        return f"⚠️ Error: {str(e)}"
-
 # ---------- Gradio UI ----------
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🤖 Advanced Salesforce AI Code Review & Chatbot")
@@ -226,14 +212,5 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         chatbot_button.click(fn=update_chatbot, inputs=[query_input, chat_state], outputs=[chatbot_output, query_input])
         clear_button.click(fn=clear_chat, inputs=None, outputs=[chatbot_output, query_input])
 
-    with gr.Tab("Create 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")
-        create_button = gr.Button("Create Record")
-        record_result = gr.Textbox(label="Result")
-        create_button.click(create_salesforce_record, inputs=[object_type, field_name, field_value], outputs=record_result)
-
 if __name__ == "__main__":
     demo.launch()