change ai model + update time estimate

Alforoan · Sep 5, 2024 · b2a7cf7 · b2a7cf7
1 parent 2e3b724
commit b2a7cf7
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 6 deletions.
diff --git a/client/src/pages/Generate.tsx b/client/src/pages/Generate.tsx
@@ -43,6 +43,7 @@ type Detail = {
   summary: string;
   format: "video" | "article";
   link?: string;
+  timeEstimate?: number;
 };
 
 type Subtopic = {
@@ -218,6 +219,9 @@ const Generate: React.FC = () => {
                     link: resourceData
                       ? resourceData.video_url || resourceData.article_url
                       : "No resource found",
+                    timeEstimate: resourceData
+                      ? resourceData.duration || 10
+                      : "No resource found",
                   };
                 }),
               };
@@ -485,7 +489,10 @@ const Generate: React.FC = () => {
 
     subtopics.forEach((subtopic, i) => {
       let details: CardDetails = {
-        timeEstimate: 40,
+        timeEstimate: subtopic.detail_list!.reduce(
+          (total, detail) => total + (detail.timeEstimate || 0),
+          0
+        ),
         notes: subtopic.summary,
         checklist: [],
       };
@@ -724,7 +731,9 @@ const Generate: React.FC = () => {
                               position="relative"
                             >
                               <Heading as="h6" size="sm">
-                                {detail.name}
+                                {detail.name}{" "}
+                                {detail.timeEstimate &&
+                                  ` ~ (${detail.timeEstimate} mins)`}
                               </Heading>
                               <Text pr={8} mt={1}>
                                 {detail.summary}

diff --git a/server/app.py b/server/app.py
@@ -655,7 +655,7 @@ def get_subtopic_details():
             Instead, rephrase the text to exclude these characters."""
 
         response = client.chat.completions.create(
-            model="gpt-3.5-turbo",
+            model="gpt-4o-mini",
             messages=[
                 {
                     "role": "system",
@@ -725,7 +725,7 @@ def refine_subtopics():
         print(f"Instructions: {instructions}")
 
         response = client.chat.completions.create(
-            model="gpt-3.5-turbo",
+            model="gpt-4o-mini",
             messages=[
                 {
                     "role": "system",
@@ -801,12 +801,13 @@ def get_youtube_tutorial(topic):
 
     if result['result']:
         video = result['result'][0]
+        duration = int(video['duration'].split(':')[0])
         return {
             'topic': topic,
             'video_url': video['link'],
             'title': video['title'],
             'description': video['descriptionSnippet'][0]['text'] if video['descriptionSnippet'] else 'No description available',
-            'duration': video['duration']
+            'duration': duration
         }
     else:
         return {
@@ -837,7 +838,8 @@ def get_bing_article(topic):
             'topic': topic,
             'article_url': item['url'],
             'title': item['name'],
-            'snippet': item['snippet']
+            'snippet': item['snippet'],
+            'duration': 10
         }
     else:
         return {