Fix import

app.py

@@ -1,11 +1,8 @@
 import streamlit as st
-import pytube as pt
 import os
 import subprocess
 import re
 import whisper
-from langchain.document_loaders import YoutubeLoader
-from langchain.text_splitter import RecursiveCharacterTextSplitter
 
 URL = 'URL'
 TEXT = 'TEXT'
@@ -19,14 +16,8 @@ model = ''
 st.title('Youtube Audio+Text')
 
 def init_state():
-    if URL not in st.session_state:
-        st.session_state[URL] = ''
-    
-    if TEXT not in st.session_state:
-        st.session_state[TEXT] = ''
-
-    if WHISPER not in st.session_state:
-        st.session_state[WHISPER] = ''
+    for state in STATES:
+        st.session_state[state] = ''
 
     if AUDIO_EXISTS not in st.session_state:
         st.session_state[AUDIO_EXISTS] = False
@@ -39,15 +30,8 @@ def clear_old_files():
         if file.endswith(".mp3") or file == 'transcript.txt':
             os.remove(file)
             print(f"Removed old files::{file}")
-
-    
-def extract_youtube_video_id(url):
-  regex = r"v=([^&]+)"
-  match = re.search(regex, url)
-  if match:
-    return match.group(1)
-  else:
-    return None
+    #Refresh audio state
+    load_audio()
 
 
 def load_whisper():
@@ -90,6 +74,7 @@ def display():
             input_submit_button = st.form_submit_button(label='Send')
 
     if input_submit_button and user_input:
+        clear_old_files()
         download()
         load_whisper()
 
@@ -100,7 +85,6 @@ def display():
                     value=st.session_state[WHISPER])
        
 def download():
-  id = extract_youtube_video_id(st.session_state[URL])
   command = [f"yt-dlp --no-config -v --extract-audio --audio-format mp3 {st.session_state[URL]} -o audio.mp3"]
   print(command)
   out = subprocess.run(command, shell=True)