Update app.py

app.py

@@ -12,16 +12,53 @@ DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
 TORCH_TYPE = torch.bfloat16 if torch.cuda.is_available() and torch.cuda.get_device_capability()[0] >= 8 else torch.float16
 
 
-DELAY_REASONS = {
-    "Step 1": ["Delay in Bead Insertion","Lack of raw material"],
-    "Step 2": ["Inner Liner Adjustment by Technician","Person rebuilding defective Tire Sections"],
-    "Step 3": ["Manual Adjustment in Ply1 apply","Technician repairing defective Tire Sections"],
-    "Step 4": ["Delay in Bead set","Lack of raw material"],
-    "Step 5": ["Delay in Turnup","Lack of raw material"],
-    "Step 6": ["Person Repairing sidewall","Person rebuilding defective Tire Sections"],
-    "Step 7": ["Delay in sidewall stitching","Lack of raw material"],
-    "Step 8": ["No person available to load Carcass","No person available to collect tire"]
-}
+def get_step_info(step_name):
+    """Returns detailed information about a manufacturing step."""
+    step_details = {
+        "Step 1": {
+            "Name": "Bead Insertion",
+            "Standard Time": "4 seconds",
+            "Analysis": "Observe the bead placement process. If the insertion exceeds 4 seconds, identify potential issues such as missing beads, technician errors, or machinery malfunction."
+        },
+        "Step 2": {
+            "Name": "Inner Liner Apply",
+            "Standard Time": "4 seconds",
+            "Analysis": "Check for manual intervention during the inner layer application. If adjustment is required, it may indicate improper alignment or issues with the layer material."
+        },
+        "Step 3": {
+            "Name": "Ply1 Apply",
+            "Standard Time": "4 seconds",
+            "Analysis": "Determine if the technician is manually adjusting the first ply. Manual intervention might suggest improper ply placement or machine misalignment."
+        },
+        "Step 4": {
+            "Name": "Bead Set",
+            "Standard Time": "8 seconds",
+            "Analysis": "Observe the bead setting process. Delays may result from bead misalignment, machine pauses, or lack of technician involvement."
+        },
+        "Step 5": {
+            "Name": "Turnup",
+            "Standard Time": "4 seconds",
+            "Analysis": "Examine the turnup step for any technician involvement or pauses in machine operation. Reasons for delays might include material misalignment or equipment issues."
+        },
+        "Step 6": {
+            "Name": "Sidewall Apply",
+            "Standard Time": "14 seconds",
+            "Analysis": "If a technician is repairing the sidewall, this may indicate material damage or improper initial application. Look for signs of excessive manual handling."
+        },
+        "Step 7": {
+            "Name": "Sidewall Stitching",
+            "Standard Time": "5 seconds",
+            "Analysis": "Observe the stitching process. Delays could occur due to machine speed inconsistencies or technician intervention for correction."
+        },
+        "Step 8": {
+            "Name": "Carcass Unload",
+            "Standard Time": "7 seconds",
+            "Analysis": "Ensure a technician is present for the carcass unload. If absent, note their return time and identify potential reasons for their absence."
+        }
+    }
+    
+    return step_details.get(step_name, {"Error": "Invalid step name. Please provide a valid step number."})
+
 
 def load_video(video_data, strategy='chat'):
     """Loads and processes video data into a format suitable for model input."""