Update app.py

app.py

@@ -161,7 +161,7 @@ import os
 import subprocess
 import sys
 # Run the .bat file before launching the app
-"""try:
+try:
     import PromptTrack
 except ImportError:
     print("PromptTrack not found. Installing...")
@@ -175,7 +175,7 @@ except ImportError:
 
 
 from PromptTrack import PromptTracker
-tracker = PromptTracker()"""
+tracker = PromptTracker()
 def process_video(video_path, prompt):
     detection_threshold=0.3
     track_thresh=0.4
@@ -186,12 +186,12 @@ def process_video(video_path, prompt):
     output_file = video_path.split('mp4')[0]+"_mot_.json"    # Tracking result
     output_file_2 = video_path.split('mp4')[0]+"_object_detection.json"    # detection results
     video_file = video_path
-    """tracker.detect_objects(video_file, prompt=prompt, nms_threshold=0.8, detection_threshold=detection_threshold, detector="OWL-VITV2")
+    tracker.detect_objects(video_file, prompt=prompt, nms_threshold=0.8, detection_threshold=detection_threshold, detector="OWL-VITV2")
     tracker.process_mot(video_file, fixed_parc=True, track_thresh=track_thresh, match_thresh=match_thresh, frame_rate=25, max_time_lost=max_time_lost, nbr_frames_fixing=nbr_frames_fixing)
-    tracker.read_video_with_mot(video_file, fps=25)"""
+    tracker.read_video_with_mot(video_file, fps=25)
     
     
-    output_video = "output.mp4"  # Placeholder for processed video
+    """output_video = "output.mp4"  # Placeholder for processed video
     output_file = "output.txt"    # Placeholder for generated file
     
     
@@ -201,7 +201,7 @@ def process_video(video_path, prompt):
     # Create an output text file with the prompt content
     with open(output_file, "w") as f:
         f.write(f"User Prompt: {prompt}\n")
-    
+    """
     return output_video, output_file
 
 # Define Gradio interface
@@ -219,33 +219,3 @@ if __name__ == "__main__":
     iface.launch(share=True)
 
 
-'''
-import gradio as gr
-import shutil
-import os
-
-def process_video(video, prompt):
-    output_video = "output.mp4"  # Placeholder for processed video
-    output_file = "output.txt"    # Placeholder for generated file
-    
-    # Copy the input video to simulate processing
-    shutil.copy(video.name, output_video)
-    
-    # Create an output text file with the prompt content
-    with open(output_file, "w") as f:
-        f.write(f"User Prompt: {prompt}\n")
-    
-    return output_video, output_file
-
-# Define Gradio interface
-iface = gr.Interface(
-    fn=process_video,
-    inputs=[gr.File(label="Upload Video"), gr.Textbox(placeholder="Enter your prompt")],
-    outputs=[gr.Video(), gr.File(label="Generated File")],
-    title="Video Processing App",
-    description="Upload a video and enter a prompt. The app will return the processed video and a generated file."
-)
-
-# Launch the app
-if __name__ == "__main__":
-    iface.launch(share=True)'''