Update app.py

app.py

@@ -6,76 +6,23 @@ from decord import cpu, VideoReader, bridge
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers import BitsAndBytesConfig
 
+
 MODEL_PATH = "THUDM/cogvlm2-llama3-caption"
 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"]
+    "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 analyze_video_step(step_name, observed_time, issues=""):
-    """
-    Analyzes a video step based on its name and observed time.
-
-    Parameters:
-    step_name (str): The name of the step.
-    observed_time (int): Observed time taken for the step (in seconds).
-    issues (str): Any specific issues noted during the analysis.
-
-    Returns:
-    str: Analysis result for the provided step.
-    """
-    match step_name:
-        case "Bead Insertion":
-            standard_time = 4
-            analysis = "Missing beads, technician errors, or machinery malfunction.Technician is unavailable at the time of bead insertion."
-        case "Inner Liner Apply":
-            standard_time = 4
-            analysis = "Manual intervention or alignment issues.If technician is manually repairing inner liner."
-        case "Ply1 Apply":
-            standard_time = 4
-            analysis = "Manual adjustment suggesting improper placement or misalignment."
-        case "Bead Set":
-            standard_time = 8
-            analysis = "Bead misalignment, machine pauses, or technician involvement."
-        case "Turnup":
-            standard_time = 4
-            analysis = "Material misalignment or equipment issues."
-        case "Sidewall Apply":
-            standard_time = 14
-            analysis = "Material damage or improper application.Technician repairing sidewall."
-        case "Sidewall Stitching":
-            standard_time = 5
-            analysis = "Machine speed inconsistencies or manual correction."
-        case "Carcass Unload":
-            standard_time = 7
-            analysis = "Absence of technician or delayed involvement.Technician not available to collect tire or load carcass"
-        case _:
-            return "Invalid step name. Please provide a valid step name."
-
-    if observed_time > standard_time:
-        return (
-            f"Step: {step_name}\n"
-            f"Standard Time: {standard_time} seconds\n"
-            f"Observed Time: {observed_time} seconds\n"
-            f"Analysis: Delay detected. Potential issues: {analysis} {issues}"
-        )
-    else:
-        return (
-            f"Step: {step_name}\n"
-            f"Standard Time: {standard_time} seconds\n"
-            f"Observed Time: {observed_time} seconds\n"
-            "Analysis: Step completed within standard time."
-        )
-
 def load_video(video_data, strategy='chat'):
     """Loads and processes video data into a format suitable for model input."""
     bridge.set_bridge('torch')
@@ -176,10 +123,12 @@ Please provide your analysis in the following format:
 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."""
 
+
+# Load model globally
 model, tokenizer = load_model()
 
-def inference(video, step_number, observed_time, issues=""):
-    """Analyzes video and additional step data for delay analysis."""
+def inference(video, step_number):
+    """Analyzes video to predict the most likely cause of delay in the selected manufacturing step."""
     try:
         if not video:
             return "Please upload a video first."
@@ -187,10 +136,9 @@ def inference(video, step_number, observed_time, issues=""):
         possible_reasons = DELAY_REASONS[step_number]
         prompt = get_analysis_prompt(step_number, possible_reasons)
         temperature = 0.8
-        video_response = predict(prompt, video, temperature, model, tokenizer)
-        step_analysis = analyze_video_step(step_number, observed_time, issues)
+        response = predict(prompt, video, temperature, model, tokenizer)
         
-        return f"Video Analysis:\n{video_response}\n\nStep Analysis:\n{step_analysis}"
+        return response
     except Exception as e:
         return f"An error occurred during analysis: {str(e)}"
 
@@ -240,6 +188,5 @@ def create_interface():
     return demo
 
 if __name__ == "__main__":
-    model, tokenizer = load_model()
     demo = create_interface()
     demo.queue().launch(share=True)