Update app.py

app.py

@@ -2,10 +2,8 @@ import datetime
 import io
 import logging
 import sys
-import requests
 import traceback
 import gradio as gr
-import pandas as pd
 from simple_salesforce import Salesforce
 
 # Configure logging
@@ -16,7 +14,7 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
-# Replace with your actual Salesforce credentials
+# Salesforce credentials (update with your credentials)
 SALESFORCE_USERNAME = "vaneshdevarapalli866@agentforce.com"
 SALESFORCE_PASSWORD = "vanesh@331"
 SALESFORCE_SECURITY_TOKEN = "VRUVbBOdG0s9Q4xy0W6DB1Y6b"
@@ -37,57 +35,50 @@ def connect_to_salesforce():
 
 sf = connect_to_salesforce()
 
-# Sample Equipment and Project entries (Name, Salesforce Record ID)
+# Your Equipment and Project picklist values
 equipment_choices = [
-    ("Bulldozer", "a0B3X00000FfPqLUAV"),
-    ("Crane", "a0B3X00000FfPqLVAV"),
-    ("Excavator", "a0B3X00000FfPqLWAV")
+    "Bulldozer", "Crane", "Excavator", "Loader", "Forklift",
+    "Backhoe", "Grader", "Scraper", "Dump Truck", "Roller"
 ]
 
 project_choices = [
-    ("Project Alpha", "a0C3X00000PqPqXUAV"),
-    ("Project Beta", "a0C3X00000PqPqYUAV"),
-    ("Project Gamma", "a0C3X00000PqPqZUAV")
+    "Project Alpha", "Project Beta", "Project Gamma", "Project Delta", "Project Epsilon",
+    "Project Zeta", "Project Theta", "Project Sigma", "Project Omega", "Project Phoenix"
 ]
 
-def get_id_from_name(name, choices):
-    for n, id_ in choices:
-        if n == name:
-            return id_
-    return None
+def call_ai_model(usage_hours, idle_hours, movement_frequency, cost_per_hour, last_maintenance_str):
+    # Placeholder for your Hugging Face AI model integration
+    # Replace this function with actual API call to your HF endpoint
+    # For now, dummy values returned:
+    return "Move", 0.9, 0.85  # suggestion, confidence (0-1), utilization_score (0-1)
 
 def process_equipment_utilization(equipment_name, project_name, usage_hours, idle_hours,
                                   movement_frequency, cost_per_hour, report_link, last_maintenance):
-    equipment_id = get_id_from_name(equipment_name, equipment_choices)
-    project_id = get_id_from_name(project_name, project_choices)
-
-    if not equipment_id or not project_id:
-        return {"error": "Could not find Salesforce record IDs for Equipment or Project. Please check names."}
-
     last_maintenance_str = last_maintenance.strftime('%Y-%m-%d') if last_maintenance else None
 
-    # Dummy AI suggestion for example
-    ai_suggestion = "Move"
-    suggestion_confidence = 90
-    utilization_score = 85
+    # Call AI model to get suggestion, confidence, and utilization score
+    ai_suggestion, suggestion_confidence, utilization_score = call_ai_model(
+        usage_hours, idle_hours, movement_frequency, cost_per_hour, last_maintenance_str or ""
+    )
 
     salesforce_record = {
-        "EquipmentID__c": equipment_id,
-        "ProjectID__c": project_id,
+        "Equipment_Name__c": equipment_name,     # Text/Picklist field in Salesforce
+        "Project_Name__c": project_name,         # Text/Picklist field in Salesforce
         "Usage_Hours__c": usage_hours,
         "Idle_Hours__c": idle_hours,
         "AI_Suggestion__c": ai_suggestion,
-        "Suggestion_Confidence__c": suggestion_confidence,
-        "Utilization_Score__c": utilization_score,
+        "Suggestion_Confidence__c": suggestion_confidence * 100,  # Store as Percent (0-100)
+        "Utilization_Score__c": utilization_score * 100,         # Store as Percent (0-100)
         "Report_Link__c": report_link,
         "Last_Maintenance__c": last_maintenance_str,
         "Dashboard_Flag__c": False
     }
+
     try:
         response = sf.Equipment_Utilization_Record__c.create(salesforce_record)
-        logger.info(f"Salesforce Equipment Utilization Record created: {response}")
+        logger.info(f"Salesforce record created: {response}")
         return {
-            "Salesforce_Record_Id": response["id"],
+            "Salesforce_Record_Id": response.get("id"),
             "status": "Success",
             "AI_Suggestion": ai_suggestion,
             "Suggestion_Confidence": suggestion_confidence,
@@ -128,16 +119,16 @@ def gradio_upload_process(equipment_name, project_name, usage_hours, idle_hours,
         return result
     except Exception as e:
         tb = traceback.format_exc()
-        logger.error(f"Unexpected error in upload process: {tb}")
-        return {"error": f"Unexpected error: {str(e)}\n{tb}"}
+        logger.error(f"Unexpected error: {tb}")
+        return {"error": f"Unexpected error: {str(e)}"}
 
 with gr.Blocks() as app:
     gr.Markdown("# Equipment Utilization Record Uploader")
-    gr.Markdown("Select Equipment and Project, enter utilization details, and save to Salesforce.")
+    gr.Markdown("Fill the form below and save the data directly to your Salesforce Equipment Utilization Record object.")
 
     with gr.Row():
-        equipment_dropdown = gr.Dropdown(choices=[name for name, _ in equipment_choices], label="Equipment")
-        project_dropdown = gr.Dropdown(choices=[name for name, _ in project_choices], label="Project")
+        equipment_dropdown = gr.Dropdown(choices=equipment_choices, label="Equipment Name")
+        project_dropdown = gr.Dropdown(choices=project_choices, label="Project Name")
 
     usage_hours = gr.Number(label="Usage Hours", value=0, minimum=0)
     idle_hours = gr.Number(label="Idle Hours", value=0, minimum=0)