Update src/app.py

src/app.py

@@ -37,7 +37,7 @@ load_dotenv()
 SPACE_ID = os.getenv("SPACE_ID")
 IS_SPACES = SPACE_ID is not None
 
-
+'''
 if IS_SPACES:
     LANCEDB_HOST_FILE = "/tmp/.lancedb"
     VIDEO_DIR = "/tmp/videos/video1"
@@ -45,9 +45,9 @@ if IS_SPACES:
 else:
     LANCEDB_HOST_FILE = "./shared_data/.lancedb"
     VIDEO_DIR = "./shared_data/videos/video1"
-    
-#LANCEDB_HOST_FILE = "./shared_data_2/.lancedb"
-#VIDEO_DIR = "./shared_data_2/videos/video1"
+'''    
+LANCEDB_HOST_FILE = "./shared_data/.lancedb"
+VIDEO_DIR = "./shared_data/videos/video1"
 TBL_NAME = "vectorstore"
 
 # Initialize components
@@ -119,17 +119,22 @@ def preprocess_and_store(youtube_url: str):
     """Download video, extract frames+metadata, embed & store in LanceDB"""
     try:
         video_url = youtube_url
+        '''
         if os.getenv("SPACE_ID"):
             video_dir = "/tmp/videos/video1"
         else:
             video_dir = "./shared_data/videos/video1"
+        '''
+        video_dir = "./shared_data/videos/video1"
         
         # Use the robust download function
-        video_filepath = download_video_robust(video_url, video_dir)
+        #video_filepath = download_video_robust(video_url, video_dir)
+        video_filepath = "./shared_data/videos/video1/Welcome back to Planet Earth.mp4"
         
         # Try to download subtitle, but don't fail if it doesn't work
         try:
-            video_transcript_filepath = download_youtube_subtitle(video_url, video_dir)
+            #video_transcript_filepath = download_youtube_subtitle(video_url, video_dir)
+            video_transcript_filepath = "./shared_data/videos/video1/generated_captions.vtt"
         except Exception as e:
             print(f"Warning: Could not download subtitles: {e}")
             # Create empty transcript file