Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import os
 import openai
 import streamlit as st
@@ -68,25 +69,39 @@ def get_youtube_transcript(url):
         st.error(f"Error fetching YouTube transcript: {str(e)}")
         return None
 
-# Function to generate notes for each chunk
+# Function to generate notes for each chunk using o1-mini model
 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}],
+        model="o1-mini",  # Using the o1-mini model
+        messages=[
+            {
+                "role": "user",
+                "content": [
+                    {"type": "text", "text": prompt}
+                ]
+            }
+        ],
         max_tokens=1000,
     )
-    return response.choices[0].message.content.strip()
+    return response['choices'][0]['message']['content'].strip()
 
-# Function to generate additional sections
+# Function to generate additional sections using o1-mini model
 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='gpt-3.5-turbo',
-        messages=[{'role': 'user', 'content': prompt}],
+        model="o1-mini",  # Using the o1-mini model
+        messages=[
+            {
+                "role": "user",
+                "content": [
+                    {"type": "text", "text": 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):