Update app.py

app.py

@@ -57,7 +57,7 @@ project_choices = [
     "Project Alpha", "Project Beta", "Project Gamma", "Project Delta", "Project Epsilon",
     "Project Zeta", "Project Theta", "Project Sigma", "Project Omega", "Project Phoenix"
 ]
-ai_suggestion_choices = ["Move", "Repair", "Replace"]  # Removed "Pause Rent"
+ai_suggestion_choices = ["Move", "Repair", "Replace"]
 
 def validate_date(last_maint):
     try:
@@ -83,7 +83,6 @@ def call_ai_model(usage, idle, freq, cost, last):
         utilization_ratio = usage / total if total > 0 else 0.0
         utilization_percent = utilization_ratio * 100
 
-        # Determine suggestion (no "Pause Rent")
         if utilization_percent < 60:
             sug = "Move"
         elif utilization_percent < 80:
@@ -100,7 +99,6 @@ def call_ai_model(usage, idle, freq, cost, last):
         base_conf *= (0.7 + 0.3 * freq_factor)
         confidence = max(0.05, min(base_conf, 1.0))
 
-        # Round before return
         return sug, round(confidence, 2), round(utilization_percent, 2)
 
     except Exception as e:
@@ -112,13 +110,10 @@ def process_equipment_utilization(equip, proj, use_h, idle_h, move_f, cost_h, la
         if not sf:
             raise ValueError("Salesforce connection is not initialized. Please check credentials and try again.")
 
-        # Validate the Last Maintenance date
         last_maint = validate_date(last_maint)
 
-        # Always run AI model to get suggestion, confidence, score
         ai_sug_generated, conf, score = call_ai_model(use_h, idle_h, move_f, cost_h, last_maint)
 
-        # Use manual suggestion if provided, else AI suggestion
         suggestion_to_use = ai_sug if ai_sug else ai_sug_generated
 
         for field, value in [("Usage Hours", use_h), ("Idle Hours", idle_h), 
@@ -136,7 +131,7 @@ def process_equipment_utilization(equip, proj, use_h, idle_h, move_f, cost_h, la
             "Usage Hours": use_h,
             "Idle Hours": idle_h,
             "Suggestion": suggestion_to_use,
-            "Confidence": conf * 100,    # Store as percentage (0-100)
+            "Confidence": conf * 100,    
             "Utilization Score": score,
             "Cost per Hour": cost_h,
             "Last Maintenance": last_maint or "N/A"
@@ -216,17 +211,17 @@ def format_batch_output(records):
         
         # Start by creating a header for each record
         record_details = [
-            f"Record {i}:",
-            f" • Record ID: {rec['Salesforce_Record_Id']}",
-            f" • Equipment Name: {summary.get('Equipment Name', 'N/A')}",
-            f" • Project: {summary.get('Project', 'N/A')}",
-            f" • Usage Hours: {summary.get('Usage Hours', 0):.2f}",
-            f" • Idle Hours: {summary.get('Idle Hours', 0):.2f}",
-            f" • Suggestion: {summary.get('Suggestion', 'N/A')}",
-            f" • Suggestion Confidence: {summary.get('Confidence', 0):.2f}%",
-            f" • Utilization Score: {summary.get('Utilization Score', 0):.2f}%",
-            f" • Cost per Hour: ₹{summary.get('Cost per Hour', 0):,.2f}",
-            f" • Last Maintenance: {summary.get('Last Maintenance', 'N/A')}"
+            f"Record {i}: {summary.get('Equipment Name', 'N/A')}",
+            f"Salesforce Record ID: {rec['Salesforce_Record_Id']}",
+            f"Equipment Name: {summary.get('Equipment Name', 'N/A')}",
+            f"Project: {summary.get('Project', 'N/A')}",
+            f"Usage Hours: {summary.get('Usage Hours', 0):.2f}",
+            f"Idle Hours: {summary.get('Idle Hours', 0):.2f}",
+            f"Suggestion: {summary.get('Suggestion', 'N/A')}",
+            f"Confidence: {summary.get('Confidence', 0):.2f}",
+            f"Utilization Score: {summary.get('Utilization Score', 0):.2f}",
+            f"Cost per Hour: ₹{summary.get('Cost per Hour', 0):,.2f}",
+            f"Last Maintenance: {summary.get('Last Maintenance', 'N/A')}"
         ]
         lines.append("\n".join(record_details))
         lines.append("\n---\n")  # Separator between records
@@ -387,7 +382,7 @@ with gr.Blocks() as app:
                 submit_btn.click(
                     fn=manual_input,
                     inputs=[equipment_dropdown, project_dropdown, usage, idle, freq, cost, last, ai_dropdown],
-                    outputs=[result_txt, report_file]  # Only PDF output here now
+                    outputs=[result_txt, report_file]  
                 )
                 clear_btn.click(lambda: ("", None), None, [result_txt, report_file])
         with gr.TabItem("CSV Upload"):
@@ -404,7 +399,7 @@ with gr.Blocks() as app:
                     outputs=[csv_output, batch_pdf]
                 )
                 clear_btn.click(
-                    lambda: (None, "", None),  # Reset file input, output markdown, and pdf file output
+                    lambda: (None, "", None),  
                     None,
                     [csv_file, csv_output, batch_pdf]
                 )