Update app.py

app.py

@@ -6,10 +6,11 @@ from pathlib import Path
 import shutil
 import whisper
 from dotenv import load_dotenv
-from zipfile import ZipFile 
+from zipfile import ZipFile
 
 load_dotenv()
 
+# Load OpenAI API key from .env
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
 @st.cache_data
@@ -41,16 +42,28 @@ def audio_to_transcript(audio_file):
 
 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
+        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']
 
+def answer_question(question, context):
+    response = openai.Completion.create(
+        engine="text-davinci-003",
+        prompt=f"Q: {question}\nContext: {context}\nAnswer:",
+        temperature=0.7,
+        max_tokens=150,
+        top_p=1,
+        frequency_penalty=0,
+        presence_penalty=0,
+    )
+    answer = response.choices[0].text
+    return answer
 
 st.markdown('# Generate and Chat with Transcripts')
 
@@ -62,30 +75,42 @@ 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")
-    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"
-        )
+    if not transcript:
+        st.error("Transcript generation failed. Please try another video or check the input.")
+    else:
+        st.header("Transcript")
+        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"
+            )
+
+st.header("Ask Questions")
+user_question = st.text_input("Ask a question:")
+if st.button("Get Answer"):
+    if transcript:
+        answer = answer_question(user_question, transcript)
+        st.success("Answer: " + answer)
+    else:
+        st.error("No transcript available. Please analyze a video first.")