Update app.py

app.py

@@ -1,24 +1,87 @@
+import datetime
+import io
+import logging
+import requests
+import traceback
+import gradio as gr
+import pandas as pd
+from simple_salesforce import Salesforce
+
+# Configure logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s',
+    handlers=[logging.FileHandler('app_log.txt'), logging.StreamHandler()]
+)
+logger = logging.getLogger(__name__)
+
+# Salesforce credentials
+SALESFORCE_USERNAME = "vaneshdevarapalli866@agentforce.com"
+SALESFORCE_PASSWORD = "vanesh@331"
+SALESFORCE_SECURITY_TOKEN = "VRUVbBOdG0s9Q4xy0W6DB1Y6b"
+
+# Connect to Salesforce
+sf = Salesforce(
+    username=SALESFORCE_USERNAME,
+    password=SALESFORCE_PASSWORD,
+    security_token=SALESFORCE_SECURITY_TOKEN,
+    domain="login"
+)
+
+# Lookup helper functions
+def get_equipment_id_by_name(equipment_name):
+    try:
+        query = f"SELECT Id FROM Equipment__c WHERE Name = '{equipment_name}' LIMIT 1"
+        result = sf.query(query)
+        if result["totalSize"] > 0:
+            return result["records"][0]["Id"]
+        else:
+            logger.error(f"No Equipment found with name: {equipment_name}")
+            return None
+    except Exception as e:
+        logger.error(f"Error fetching Equipment ID: {e}")
+        return None
+
+def get_project_id_by_name(project_name):
+    try:
+        query = f"SELECT Id FROM Project__c WHERE Name = '{project_name}' LIMIT 1"
+        result = sf.query(query)
+        if result["totalSize"] > 0:
+            return result["records"][0]["Id"]
+        else:
+            logger.error(f"No Project found with name: {project_name}")
+            return None
+    except Exception as e:
+        logger.error(f"Error fetching Project ID: {e}")
+        return None
+
+# CSV generation
+def generate_csv_report(data_dict):
+    df = pd.DataFrame([data_dict])
+    buffer = io.StringIO()
+    df.to_csv(buffer, index=False)
+    buffer.seek(0)
+    return buffer.getvalue()
+
+# Record creation
 def process_equipment_utilization(equipment_id, project_id, usage_hours, idle_hours,
                                   movement_frequency, cost_per_hour, report_link, last_maintenance):
     last_maintenance_str = last_maintenance.strftime('%Y-%m-%d') if last_maintenance else None
 
-    # Lookup Equipment ID by Name
     equipment_sf_id = get_equipment_id_by_name(equipment_id)
     if not equipment_sf_id:
         return {"error": f"Equipment name '{equipment_id}' not found in Salesforce."}
 
-    # Lookup Project ID by Name
     project_sf_id = get_project_id_by_name(project_id)
     if not project_sf_id:
         return {"error": f"Project name '{project_id}' not found in Salesforce."}
 
-    # Static values instead of AI model
+    # Placeholder AI values (since endpoint removed)
     ai_suggestion = "Move"
-    suggestion_confidence = 85
-    utilization_score = 78
+    suggestion_confidence = 90
+    utilization_score = 88
 
-    # Create record in your custom object
-    record = {
+    salesforce_record = {
         "Equipment_ID__c": equipment_sf_id,
         "Project_ID__c": project_sf_id,
         "Usage_Hours__c": usage_hours,
@@ -31,16 +94,15 @@ def process_equipment_utilization(equipment_id, project_id, usage_hours, idle_ho
         "Report_Link__c": report_link
     }
 
-    try:
-        response = sf.Equipment_Utilization_Record__c.create(record)
-        logger.info(f"Record created: {response}")
-        return {
-            "Salesforce_Record_Id": response["id"],
-            "status": "Success",
-            "AI_Suggestion": ai_suggestion,
-            "Suggestion_Confidence": suggestion_confidence,
-            "Utilization_Score": utilization_score
-        }
-    except Exception as e:
-        logger.error(f"Insert error: {e}")
-        return {"error": str(e)}
+    response = sf.Equipment_Utilization_Record__c.create(salesforce_record)
+    logger.info(f"Salesforce Equipment Utilization Record created: {response}")
+    csv_content = generate_csv_report(salesforce_record)
+    return {
+        "Salesforce_Record_Id": response["id"],
+        "status": "Success",
+        "AI_Suggestion": ai_suggestion,
+        "Suggestion_Confidence": suggestion_confidence,
+        "Utilization_Score": utilization_score,
+        "Report_Link": report_link,
+        "CSV_Report": csv_content
+    }