Update app.py

app.py

@@ -62,17 +62,15 @@ def get_youtube_transcript(url):
         # Fetch the actual transcript data
         transcript_data = transcript.fetch()
         # Combine transcript texts
-        transcription_text = "\n".join(
-            [f"[{entry['start']:.2f}-{entry['start']+entry['duration']:.2f}] {entry['text']}" for entry in transcript_data]
-        )
+        transcription_text = "\n".join([entry['text'] for entry in transcript_data])
         return transcription_text
     except Exception as e:
         st.error(f"Error fetching YouTube transcript: {str(e)}")
         return None
 
-# Function to generate summary for each chunk
-def generate_summary(text):
-    prompt = f"Summarize the following text into 500-800 words, focusing on key points:\n\n{text}"
+# Function to generate notes for each chunk
+def generate_notes(text):
+    prompt = f"Create comprehensive notes in bullet points from the following text:\n\n{text}"
     response = openai.ChatCompletion.create(
         model='gpt-3.5-turbo',
         messages=[{'role': 'user', 'content': prompt}],
@@ -80,16 +78,6 @@ def generate_summary(text):
     )
     return response.choices[0].message.content.strip()
 
-# Function to chat with AI
-def chat_with_ai(question, context):
-    prompt = f"Based on the following content:\n\n{context}\n\nAnswer the question:\n{question}"
-    response = openai.ChatCompletion.create(
-        model='gpt-3.5-turbo',
-        messages=[{'role': 'user', 'content': prompt}],
-        max_tokens=500,
-    )
-    return response.choices[0].message.content.strip()
-
 # Main app
 def main():
     st.set_page_config(layout="wide")
@@ -99,9 +87,9 @@ def main():
     st.sidebar.header("Upload your file:")
     input_type = st.sidebar.selectbox("Select Input Type", ["Audio File", "PDF Document", "YouTube URL"])
     st.sidebar.markdown("### Steps to Use the Tool:")
-    st.sidebar.markdown("1. Upload an audio file (25MB max), PDF, or YouTube URL for summary.")
-    st.sidebar.markdown("2. Click on 'Generate Notes' to get an AI-transcribed summary.")
-    st.sidebar.markdown("3. Use 'Chat with AI' for further questions.")
+    st.sidebar.markdown("1. Upload an audio file (25MB max), PDF, or YouTube URL for notes.")
+    st.sidebar.markdown("2. Click on 'Generate Notes' to get AI-generated notes.")
+    st.sidebar.markdown("3. Use 'Download Notes as PDF' to save them locally.")
 
     # File uploader or URL input based on selected type
     audio_input = pdf_input = youtube_input = None
@@ -113,19 +101,17 @@ def main():
         youtube_input = st.sidebar.text_input("Enter YouTube URL (must have subtitles enabled)", key="youtube")
 
     # Main section for processing and results
-    if st.button("Generate Notes", key="generate_summary"):
+    if st.button("Generate Notes", key="generate_notes"):
         transcription_text = ""
         if input_type == "Audio File" and audio_input:
             transcription = transcribe_audio(audio_input)
             if transcription:
-                transcription_text = "\n".join(
-                    [f"[{seg['start']:.2f}-{seg['end']:.2f}] {seg['text']}" for seg in transcription['segments']]
-                )
+                transcription_text = "\n".join([seg['text'] for seg in transcription['segments']])
             else:
                 st.error("Transcription failed.")
         elif input_type == "PDF Document" and pdf_input:
             chunks = extract_text_from_pdf(pdf_input)
-            summaries = [generate_summary(chunk) for chunk in chunks]
+            summaries = [generate_notes(chunk) for chunk in chunks]
             transcription_text = "\n".join(summaries)
         elif input_type == "YouTube URL" and youtube_input:
             transcription_text = get_youtube_transcript(youtube_input)
@@ -135,43 +121,30 @@ def main():
             st.error("Please provide valid input.")
 
         if transcription_text:
-            st.session_state['transcription'] = transcription_text
-            summary = generate_summary(transcription_text)
-            st.session_state['summary'] = summary
-
-    # Display transcription and summary side by side
-    if 'transcription' in st.session_state and 'summary' in st.session_state:
-        st.markdown("---")
-        col1, col2 = st.columns([1, 1])
+            st.session_state['summary'] = transcription_text
 
-        with col1:
-            st.subheader("Transcription with Timestamps")
-            st.text_area("", st.session_state['transcription'], height=300)
-
-        with col2:
-            st.subheader("Generated Notes")
-            st.text_area("", st.session_state['summary'], height=300)
-
-        # Chat with AI at the bottom
+    # Display generated notes
+    if 'summary' in st.session_state:
         st.markdown("---")
-        st.subheader("Chat with AI")
-        question = st.text_input("Ask a question about the content:")
-        if st.button("Save Notes to PDF"):
-            pdf = FPDF()
-            pdf.add_page()
-            pdf.set_auto_page_break(auto=True, margin=15)
-            pdf.set_font("Arial", size=12)
-            for line in st.session_state['summary'].split('\n'):
-                pdf.multi_cell(0, 10, line)
-            pdf_output = io.BytesIO()
-            pdf.output(pdf_output)
-            pdf_output.seek(0)
-            st.download_button(
-                label="Download Notes as PDF",
-                data=pdf_output,
-                file_name="notes.pdf",
-                mime="application/pdf"
-            )
+        st.subheader("Generated Notes")
+        st.text_area("", st.session_state['summary'], height=400)
+
+        # Button to save notes as PDF
+        pdf = FPDF()
+        pdf.add_page()
+        pdf.set_auto_page_break(auto=True, margin=15)
+        pdf.set_font("Arial", size=12)
+        for line in st.session_state['summary'].split('\n'):
+            pdf.multi_cell(0, 10, line)
+        pdf_output = io.BytesIO()
+        pdf.output(pdf_output)
+        pdf_output.seek(0)
+        st.download_button(
+            label="Download Notes as PDF",
+            data=pdf_output,
+            file_name="notes.pdf",
+            mime="application/pdf"
+        )
 
 if __name__ == "__main__":
     main()