Upload 2 files

app.py

@@ -0,0 +1,91 @@
+import streamlit as st
+import openai
+from pytube import YouTube
+import os
+from pathlib import Path
+import shutil
+import whisper
+from dotenv import load_dotenv
+from zipfile import ZipFile 
+
+load_dotenv()
+
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+@st.cache_data
+def load_model():
+    model = whisper.load_model("base")
+    return model
+
+def save_audio(url):
+    yt = YouTube(url)
+    video = yt.streams.filter(only_audio=True).first()
+    out_file = video.download()
+    base, ext = os.path.splitext(out_file)
+    file_name = base + '.mp3'
+    try:
+        os.rename(out_file, file_name)
+    except WindowsError:
+        os.remove(file_name)
+        os.rename(out_file, file_name)
+    audio_filename = Path(file_name).stem+'.mp3'
+    print(yt.title + " Has been successfully downloaded")
+    print(file_name)
+    return yt.title, audio_filename
+
+def audio_to_transcript(audio_file):
+    model = load_model()
+    result = model.transcribe(audio_file)
+    transcript = result["text"]
+    return transcript
+
+def text_to_news_article(text):
+    response = openai.Completion.create(
+    model="text-davinci-003",
+    prompt="Write a news article in 500 words from the below text:\n"+text,
+    temperature=0.7,
+    max_tokens=600,
+    top_p=1,
+    frequency_penalty=0,
+    presence_penalty=0
+    )
+    return response['choices'][0]['text']
+
+
+st.markdown('# 📝 **News Article Generator App**')
+
+st.header('Input the Video URL')
+
+url_link = st.text_input('Enter URL of YouTube video:')
+
+if st.checkbox('Start Analysis'):
+    video_title, audio_filename = save_audio(url_link)
+    st.audio(audio_filename)
+    transcript = audio_to_transcript(audio_filename)
+    st.header("Transcript are getting generated...")
+    st.success(transcript)
+    st.header("News Article")
+    result = text_to_news_article(transcript)
+    st.success(result)
+    
+    #save the files
+    transcript_txt = open('transcript.txt', 'w')
+    transcript_txt.write(transcript)
+    transcript_txt.close()  
+    
+    article_txt = open('article.txt', 'w')
+    article_txt.write(result) 
+    article_txt.close() 
+    
+    zip_file = ZipFile('output.zip', 'w')
+    zip_file.write('transcript.txt')
+    zip_file.write('article.txt')
+    zip_file.close()
+    
+    with open("output.zip", "rb") as zip_download:
+        btn = st.download_button(
+            label="Download ZIP",
+            data=zip_download,
+            file_name="output.zip",
+            mime="application/zip"
+        )

requirements.txt

@@ -0,0 +1,82 @@
+aiohttp==3.8.5
+aiosignal==1.3.1
+altair==5.1.2
+async-timeout==4.0.3
+attrs==23.1.0
+blinker==1.6.2
+cachetools==5.3.1
+certifi==2023.7.22
+charset-normalizer==3.3.0
+click==8.1.7
+cmake==3.27.6
+filelock==3.12.4
+frozenlist==1.4.0
+gitdb==4.0.10
+GitPython==3.1.37
+idna==3.4
+importlib-metadata==6.8.0
+Jinja2==3.1.2
+jsonschema==4.19.1
+jsonschema-specifications==2023.7.1
+lit==17.0.2
+llvmlite==0.41.0
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+mdurl==0.1.2
+more-itertools==10.1.0
+mpmath==1.3.0
+multidict==6.0.4
+networkx==3.1
+numba==0.58.0
+numpy==1.25.2
+nvidia-cublas-cu11==11.10.3.66
+nvidia-cuda-cupti-cu11==11.7.101
+nvidia-cuda-nvrtc-cu11==11.7.99
+nvidia-cuda-runtime-cu11==11.7.99
+nvidia-cudnn-cu11==8.5.0.96
+nvidia-cufft-cu11==10.9.0.58
+nvidia-curand-cu11==10.2.10.91
+nvidia-cusolver-cu11==11.4.0.1
+nvidia-cusparse-cu11==11.7.4.91
+nvidia-nccl-cu11==2.14.3
+nvidia-nvtx-cu11==11.7.91
+openai==0.28.1
+openai-whisper @ git+https://github.com/openai/whisper.git@0a60fcaa9b86748389a656aa013c416030287d47
+packaging==23.2
+pandas==2.1.1
+Pillow==10.0.1
+protobuf==4.24.4
+pyarrow==13.0.0
+pydeck==0.8.1b0
+pydub==0.25.1
+Pygments==2.16.1
+python-dateutil==2.8.2
+python-dotenv==1.0.0
+pytube==15.0.0
+pytz==2023.3.post1
+referencing==0.30.2
+regex==2023.10.3
+requests==2.31.0
+rich==13.6.0
+rpds-py==0.10.3
+six==1.16.0
+smmap==5.0.1
+streamlit==1.27.2
+sympy==1.12
+tenacity==8.2.3
+tiktoken==0.3.3
+toml==0.10.2
+toolz==0.12.0
+torch==2.0.1
+tornado==6.3.3
+tqdm==4.66.1
+triton==2.0.0
+typing_extensions==4.8.0
+tzdata==2023.3
+tzlocal==5.1
+urllib3==2.0.6
+validators==0.22.0
+watchdog==3.0.0
+whisper==1.1.10
+yarl==1.9.2
+zipp==3.17.0