Update app.py

app.py

@@ -53,38 +53,41 @@ def process_input(input_file):
 
     if ext in ['.mp4', '.avi', '.mov']:
         cap = cv2.VideoCapture(file_path)
-        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+
+        if not cap.isOpened():
+            return None, "Could not open video file", ""
+
         fps = cap.get(cv2.CAP_PROP_FPS)
-        width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
-        height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+        width = 640
+        height = 640
 
-        # Output path
-        timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
-        output_path = f"annotated_{timestamp}.mp4"
-        out = cv2.VideoWriter(output_path, fourcc, fps, (640, 640))
+        # Generate safe output path
+        output_path = f"annotated_output.mp4"
+        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+        out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
 
-        while cap.isOpened():
+        while True:
             ret, frame = cap.read()
             if not ret:
                 break
-            annotated = annotate_frame(frame)
-            annotated_resized = cv2.resize(annotated, (640, 640))
-            out.write(cv2.cvtColor(annotated_resized, cv2.COLOR_RGB2BGR))
+            frame_resized = cv2.resize(frame, (640, 640))
+            annotated_frame = annotate_frame(frame_resized)
+            bgr_frame = cv2.cvtColor(annotated_frame, cv2.COLOR_RGB2BGR)
+            out.write(bgr_frame)
 
         cap.release()
         out.release()
 
-        return output_path, "", ""
+        return output_path, "Bangla text in video (see frames)", "OCR confidence displayed on frames"
 
     else:
-        # Image case
+        # Handle image
         frame = cv2.imread(file_path)
         if frame is None:
-            return None, "Invalid image", ""
+            return None, "Invalid image file", ""
 
         annotated = annotate_frame(frame)
-        pil_img = Image.fromarray(annotated)
-        return pil_img, "", ""
+        return Image.fromarray(annotated), "Bangla text in image", ""
 
 
 interface = gr.Interface(