add download button for logs

.gitignore

@@ -2,4 +2,7 @@ __pycache__
 pose_env_1
 testing
 vit_env
+models
 test_vit_model.ipynb
+logs/*.txt
+videos/*.mp4

app.py

@@ -104,12 +104,15 @@ with gr.Blocks() as demo:
         output_merged = gr.Video(show_download_button=True)
 
     with gr.Row():
-        general_log = gr.TextArea(lines=10, max_lines=9999, label="Error log")
+        general_log = gr.TextArea(lines=10, label="Error log", scale=3)
+        text_log=gr.File(label="Download logs and settings", scale=1)
+
+  
 
     gr_button.click(
         fn=video_identity,
         inputs=[dtw_mean, dtw_filter, angles_sensitive, angles_common, angles_insensitive, trigger_state, input_teacher, input_student],
-        outputs=[output_merged, general_log]
+        outputs=[output_merged, general_log, text_log]
     )
 
 

main_func.py

@@ -21,6 +21,7 @@ from dtaidistance import dtw
 import numpy as np
 from scipy.signal import savgol_filter
 from scipy.stats import mstats
+import datetime
 from datetime import timedelta
 import cv2
 
@@ -31,15 +32,15 @@ def video_identity(dtw_mean, dtw_filter, angles_sensitive, angles_common, angles
 
     detection_result_teacher = predict_keypoints_vitpose(
     video_path=video_teacher,
-    model_path="models/vitpose-l-wholebody.pth",
-    model_name="l",
+    model_path="models/vitpose-b-wholebody.pth",
+    model_name="b",
     detector_path="models/yolov8s.pt"
     )
 
     detection_result_student = predict_keypoints_vitpose(
         video_path=video_student,
-        model_path="models/vitpose-l-wholebody.pth",
-        model_name="l",
+        model_path="models/vitpose-b-wholebody.pth",
+        model_name="b",
         detector_path="models/yolov8s.pt"
     )
 
@@ -143,10 +144,13 @@ def video_identity(dtw_mean, dtw_filter, angles_sensitive, angles_common, angles
 
 
     concat_video = np.concatenate((save_teacher_frames_resized, save_student_frames_resized), axis=2)
-
     concat_video = np.array(concat_video)
 
-    out = cv2.VideoWriter("videos/concat_modified.mp4", cv2.VideoWriter_fourcc(*'mp4v'), 30, (1280*2, 720))
+    current_time = datetime.datetime.now()
+    timestamp_str = current_time.strftime("%Y_%m-%d_%H_%M_%S")
+    video_path = f"videos/pose_{timestamp_str}.mp4"
+
+    out = cv2.VideoWriter(video_path, cv2.VideoWriter_fourcc(*'mp4v'), 30, (1280*2, 720))
     for frame in concat_video:
         out.write(frame)
     out.release()
@@ -164,6 +168,36 @@ def video_identity(dtw_mean, dtw_filter, angles_sensitive, angles_common, angles
         
         general_summary.append(f"{comment} on frame {frame}. Video time: {str(timedelta(seconds=total_seconds))[3:-4]}")
 
+
     general_summary = "\n".join(general_summary)
 
-    return "videos/concat_modified.mp4", general_summary
+    log_path = f"logs/log_{timestamp_str}.txt"
+
+
+    content = f"""
+Settings:
+
+Dynamic Time Warping:
+- Winsorize mean: {dtw_mean}
+- Savitzky-Golay Filter: {dtw_filter}
+
+Thresholds:
+- Sensitive: {angles_sensitive}
+- Standart: {angles_common}
+- Insensitive: {angles_insensitive}
+
+Patience:
+- trigger count: {trigger_state}
+
+
+Error logs:
+
+{general_summary}
+"""
+
+    with open(log_path, "w") as file:
+        file.write(content)
+
+
+
+    return video_path, general_summary, log_path

utils.py

@@ -393,10 +393,10 @@ def download_file(url, save_path):
 
 def check_and_download_models():
 
-    vit_model_url = "https://huggingface.co/JunkyByte/easy_ViTPose/resolve/main/torch/wholebody/vitpose-l-wholebody.pth?download=true"
+    vit_model_url = "https://huggingface.co/JunkyByte/easy_ViTPose/resolve/main/torch/wholebody/vitpose-b-wholebody.pth?download=true"
     yolo_model_url = "https://huggingface.co/JunkyByte/easy_ViTPose/resolve/main/yolov8/yolov8s.pt?download=true"
 
-    vit_model_path = "models/vitpose-l-wholebody.pth"
+    vit_model_path = "models/vitpose-b-wholebody.pth"
 
     yolo_model_path = "models/yolov8s.pt"