Rename SUPPORTED_FILE_TYPES to SUPPORTED_AV_FILE_TYPES

Author: punitarani

app.py

@@ -5,7 +5,7 @@
 
 from rabbithole import summarize_document
 from rabbithole.loader import load_file
-from rabbithole.mp3 import SUPPORTED_FILE_TYPES
+from rabbithole.mp3 import SUPPORTED_AV_FILE_TYPES
 
 # Global variables
 results = {}
@@ -36,7 +36,7 @@ def run_summarization(document: list[Document], doc_name: str):
 
 st.title("RabbitHole")
 
-uploaded_files = st.file_uploader("Upload content", type=["docx", "pdf", "txt", *SUPPORTED_FILE_TYPES],
+uploaded_files = st.file_uploader("Upload content", type=["docx", "pdf", "txt", *SUPPORTED_AV_FILE_TYPES],
                                   accept_multiple_files=True)
 
 if st.button("Summarize"):

rabbithole/loader.py

@@ -6,7 +6,7 @@
 from langchain.text_splitter import CharacterTextSplitter
 from streamlit.runtime.uploaded_file_manager import UploadedFile
 
-from rabbithole.mp3 import SUPPORTED_FILE_TYPES, convert_to_mp3
+from rabbithole.mp3 import SUPPORTED_AV_FILE_TYPES, convert_to_mp3
 from rabbithole.transcribe import transcribe
 
 
@@ -47,7 +47,7 @@ def load_file(file: UploadedFile) -> list[Document]:
         return TextLoader(file_path=temp_file).load_and_split(text_splitter=text_splitter)
 
     # Handle Audio and Video files
-    elif file.name.endswith(SUPPORTED_FILE_TYPES):
+    elif file.name.endswith(SUPPORTED_AV_FILE_TYPES):
         temp_file = save_to_temp_file(file)
 
         # Convert to mp3 and transcribe

rabbithole/mp3.py

@@ -6,7 +6,7 @@
 from pydub import AudioSegment
 from tqdm import tqdm
 
-SUPPORTED_FILE_TYPES = (
+SUPPORTED_AV_FILE_TYPES = (
     # Video formats
     "mp4", "mkv", "webm", "flv", "avi", "mov", "wmv",
     # Audio formats
@@ -26,7 +26,7 @@ def convert_to_mp3(filepath: str) -> str:
         return filepath
 
     # Check if the file type is supported
-    if filepath.rsplit('.', 1)[-1] not in SUPPORTED_FILE_TYPES:
+    if filepath.rsplit('.', 1)[-1] not in SUPPORTED_AV_FILE_TYPES:
         raise ValueError(f"Unsupported file type: {filepath.rsplit('.', 1)[-1]}")
 
     # Load video or audio file