Update app.py

app.py

@@ -1,13 +1,37 @@
+import subprocess as sp
+
+sp.check_call(['pip', 'install', 'pydrive'])
+sp.check_call(['pip', 'install', 'transformers'])
+sp.check_call(['pip', 'install', 'requests'])
 import os
 import json
-import streamlit as st
+import subprocess as sp
+import requests
 from pydrive.auth import GoogleAuth
 from pydrive.drive import GoogleDrive
 from oauth2client.service_account import ServiceAccountCredentials
+import gradio as gr
+
+# Fetching config from Hugging Face Spaces
+url = "https://huggingface.co/spaces/CS482Project/Milestone_5-Search_UI_and_pitch_video_voiceover/raw/main/config.json"
+
+response = requests.get(url)
+if response.status_code == 200:
+    config = response.json()
+    drive_config = config.get('drive_config')
+    drive_folder_id = config.get('drive_folder_id')
+    if drive_config and drive_folder_id:
+        print(drive_config)
+        print(drive_folder_id)
+    else:
+        print("Drive config or folder ID not found in the JSON.")
+else:
+    print("Failed to fetch data. Status code:", response.status_code)
 
-# Function to download videos from Drive
 def download_videos_from_drive(drive, folder_id, save_directory, video_names):
+    print("Fetching video list from Google Drive...")
     file_list = drive.ListFile({'q': f"'{folder_id}' in parents and trashed=false"}).GetList()
+    print("Total files in Drive folder:", len(file_list))  # Print total files in the Drive folder
     os.makedirs(save_directory, exist_ok=True)
     found_videos = []
 
@@ -19,22 +43,46 @@ def download_videos_from_drive(drive, folder_id, save_directory, video_names):
 
     return found_videos
 
-# Authenticate with Google Drive
+def view_videos(folder_id, video_names):
+    video_directory = "Videos"
+    os.makedirs(video_directory, exist_ok=True)
+    
+    print("Downloading videos from Google Drive...")
+    found_videos = download_videos_from_drive(drive, folder_id, video_directory, video_names)
+    print("Videos downloaded:", found_videos)
+    return found_videos
+
+def display_text(text):
+    relevant_videos = ['Memphis_Police_Protests,_Super_Bowl_LVII_Matchup_Set_|_NPR_News_Now.mp4', "Maui's_Death_Toll_Rises_To_111_|_NPR_News_Now.mp4", 'Manhunt_Continues_For_Suspect_In_Maine_Shootings_|_NPR_News_Now.mp4', 'Jordan_Calls_For_3rd_Vote_Friday_On_His_Bid_For_House_Speaker_|_NPR_News_Now.mp4', 'Israel_Gives_1M_Civilians_In_Northern_Gaza_24_Hours_To_Evacuate_|_NPR_News_Now.mp4']  # Replace with video names  
+    videos_from_drive = view_videos(drive_folder_id, relevant_videos)  
+    return videos_from_drive
+
+# Authentication with Google Drive
 gauth = GoogleAuth()
-# Add authentication steps here using 'drive_config'
+gauth.credentials = ServiceAccountCredentials.from_json_keyfile_dict(drive_config,
+                                                                    ['https://www.googleapis.com/auth/drive'])
 drive = GoogleDrive(gauth)
 
-# Streamlit app
-st.title("Video Viewer")
-st.markdown("Enter text to fetch and display videos from Google Drive")
-
-text_input = st.text_input("Enter text:")
-relevant_videos = ['video1.mp4', 'video2.mp4', 'video3.mp4', 'video4.mp4', 'video5.mp4']  # Replace with actual video names
+# Gradio Interface
+def display_videos(text):
+    videos = display_text(text)
+    video_tags = ""
+    for video in videos:
+        abs_video_path = os.path.abspath(video)
+        if os.path.exists(abs_video_path):
+            video_tag = f"<video width='320' height='240' controls><source src='file://{abs_video_path}' type='video/mp4'></video><br>"
+            print("Generated video tag:", video_tag)  # Print generated video tag
+            video_tags += video_tag
+        else:
+            print(f"Video file '{video}' not found.")
+    return video_tags
 
-if st.button("Fetch Videos"):
-    video_directory = "Videos"
-    videos = download_videos_from_drive(drive, drive_folder_id, video_directory, relevant_videos)
-    
-    for video in videos[:5]:  # Display the first five videos
-        st.video(video)
+iface = gr.Interface(
+    fn=display_videos,
+    inputs="text",
+    outputs="html",
+    title="Video Viewer",
+    description="Enter text to fetch and display videos from Google Drive"
+)
 
+iface.launch()