Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import os
 import openai
 import streamlit as st
@@ -69,39 +68,25 @@ def get_youtube_transcript(url):
         st.error(f"Error fetching YouTube transcript: {str(e)}")
         return None
 
-# Function to generate notes for each chunk using o1-mini model
+# 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="o1-mini",  # Using the o1-mini model
-        messages=[
-            {
-                "role": "user",
-                "content": [
-                    {"type": "text", "text": prompt}
-                ]
-            }
-        ],
+        model='gpt-3.5-turbo',
+        messages=[{'role': 'user', 'content': prompt}],
         max_tokens=1000,
     )
-    return response['choices'][0]['message']['content'].strip()
+    return response.choices[0].message.content.strip()
 
-# Function to generate additional sections using o1-mini model
+# Function to generate additional sections
 def generate_section(title, text):
     prompt = f"Generate a section titled '{title}' with 3-6 sentences based on the following text:\n\n{text}"
     response = openai.ChatCompletion.create(
-        model="o1-mini",  # Using the o1-mini model
-        messages=[
-            {
-                "role": "user",
-                "content": [
-                    {"type": "text", "text": prompt}
-                ]
-            }
-        ],
+        model='gpt-3.5-turbo',
+        messages=[{'role': 'user', 'content': prompt}],
         max_tokens=500,
     )
-    return response['choices'][0]['message']['content'].strip()
+    return response.choices[0].message.content.strip()
 
 # Function to create PowerPoint presentation
 def create_presentation(summary, key_concepts, key_takeaways, case_studies, glossary, faqs):
@@ -277,4 +262,4 @@ def main():
         )
 
 if __name__ == "__main__":
-    main()
+    main()