Update app.py

app.py

@@ -108,7 +108,7 @@ def load_video(video_data, strategy='chat'):
     bridge.set_bridge('torch')
     num_frames = 24
     
-    if isinstance(video_data, str): 
+    if isinstance(video_data, str):
         decord_vr = VideoReader(video_data, ctx=cpu(0))
     else:  
         decord_vr = VideoReader(io.BytesIO(video_data), ctx=cpu(0))
@@ -193,30 +193,39 @@ def get_analysis_prompt(step_number):
     
     step_name = step_info["Name"]
     standard_time = step_info["Standard Time"]
-    substeps = step_info["Video_substeps_expected"]
-    delay_reasons = DELAY_REASONS.get(f"Step {step_number}", ["No specific reasons provided."])
+    analysis = step_info["Analysis"]
 
-    substeps_text = "\n".join([f"- {time}: {action}" for time, action in substeps.items()])
-    reasons_text = "\n".join([f"- {reason}" for reason in delay_reasons])
-    
     return f"""
-You are an AI expert system analyzing manufacturing delays in tire production. Below are the details:
-Step: {step_number} - {step_name}
-Standard Time: {standard_time}
-Substeps Expected in Video:
-{substeps_text}
-
-Potential Delay Reasons:
-{reasons_text}
+You are an AI expert system specialized in analyzing manufacturing processes and identifying production delays in tire manufacturing. Your role is to accurately classify delay reasons based on visual evidence from production line footage.
+Task Context:
+You are analyzing video footage from Step {step_number} of a tire manufacturing process where a delay has been detected. The step is called {step_name}, and its standard time is {standard_time}.
+Required Analysis:
+Carefully observe the video for visual cues indicating production interruption.
+- If no person is visible in any of the frames, the reason probably might be due to their absence.
+- If a person is visible in the video and is observed touching and modifying the layers of the tire, it indicates an issue with tire patching, and the person might be repairing it.
+- Compare observed evidence against the following possible delay reasons: 
+  - {analysis}
+Following are the subactivities needs to happen in this step.
 
-Task: Analyze the provided video to identify the delay reason. Use the following format:
-1. **Selected Reason:** [Choose the most likely reason from the list above]
-2. **Visual Evidence:** [Describe specific visual cues from the video]
-3. **Reasoning:** [Explain why the selected reason fits the observed evidence]
-4. **Alternative Reasons:** [Briefly explain why other reasons are less likely]
-
-Important: Focus only on observable details in the video. If no delay is evident, state 'No delay observed.'"""
+{get_step_info(step_number)}
 
+Please provide your output in the following format:
+Output_Examples = {
+     ["Delay in Bead Insertion", "Lack of raw material"],
+     ["Inner Liner Adjustment by Technician", "Person rebuilding defective Tire Sections"],
+     ["Manual Adjustment in Ply1 Apply", "Technician repairing defective Tire Sections"],
+     ["Delay in Bead Set", "Lack of raw material"],
+     ["Delay in Turnup", "Lack of raw material"],
+     ["Person Repairing Sidewall", "Person rebuilding defective Tire Sections"],
+     ["Delay in Sidewall Stitching", "Lack of raw material"],
+     ["No person available to load Carcass", "No person available to collect tire"]
+}
+1. **Selected Reason:** [State the most likely reason from the given options]
+2. **Visual Evidence:** [Describe specific visual cues that support your selection]
+3. **Reasoning:** [Explain why this reason best matches the observed evidence]
+4. **Alternative Analysis:** [Brief explanation of why other possible reasons are less likely]
+Important: Base your analysis solely on visual evidence from the video. Focus on concrete, observable details rather than assumptions. Clearly state if no person or specific activity is observed.
+"""