From 9190d9cdb27677974bb0ed967de949c854b6ddf4 Mon Sep 17 00:00:00 2001
From: Asad Ali <asad.ali@arbisoft.com>
Date: Fri, 29 Nov 2024 12:20:58 +0500
Subject: [PATCH] GPT and transcripts

---
 src/ol_openedx_chat/block.py                  | 27 ++++++++++++++++++-
 .../static/html/student_view.html             |  6 +++++
 2 files changed, 32 insertions(+), 1 deletion(-)

diff --git a/src/ol_openedx_chat/block.py b/src/ol_openedx_chat/block.py
index 53cd0b18..15d9f016 100644
--- a/src/ol_openedx_chat/block.py
+++ b/src/ol_openedx_chat/block.py
@@ -4,6 +4,11 @@
 from xblock.core import XBlockAside, XBlock
 from webob.response import Response
 
+from xmodule.video_block.transcripts_utils import get_transcript_from_contentstore
+import openai
+
+openai.api_key = "<API_KEY"
+
 BLOCK_PROBLEM_CATEGORY = "problem"
 MULTIPLE_CHOICE_TYPE = "multiplechoiceresponse"
 
@@ -43,7 +48,24 @@ def student_view_aside(self, block, context=None):  # noqa: ARG002
         Renders the aside contents for the student view
         """  # noqa: D401
         fragment = Fragment("")
-        breakpoint()
+        if getattr(block, "category", None) == "video":
+            content, filename, mimetype = get_transcript_from_contentstore(block, 'en', 'txt', block.get_transcripts_info())
+            prompt = (
+                "Summarize the following video transcript into a concise, informative summary:\n\n"
+                f"{content}"
+            )
+            response = openai.ChatCompletion.create(
+                model="gpt-4o-mini",
+                messages=[
+                    {"role": "system", "content": "You are an expert at summarizing content."},
+                    {"role": "user", "content": prompt},
+                ],
+                max_tokens=200,
+                temperature=0.5,
+            )
+            print("\n\n\n", response["choices"][0]["message"]["content"], "\n\n\n")
+            return fragment
+
         fragment.add_content(render_template("static/html/student_view.html", {"block_key": self.scope_ids.usage_id.usage_key.block_id}))
         fragment.add_css(get_resource_bytes("static/css/ai_chat.css"))
         fragment.add_javascript(get_resource_bytes("static/js/ai_chat.js"))
@@ -77,6 +99,9 @@ def should_apply_to_block(cls, block):
         instances, the problem type of the given block needs to be retrieved in
         different ways.
         """  # noqa: D401
+        if getattr(block, "category", None) == 'video':
+            return True
+
         if getattr(block, "category", None) != BLOCK_PROBLEM_CATEGORY:
             return False
         block_problem_types = None
diff --git a/src/ol_openedx_chat/static/html/student_view.html b/src/ol_openedx_chat/static/html/student_view.html
index 3766423b..064c550e 100644
--- a/src/ol_openedx_chat/static/html/student_view.html
+++ b/src/ol_openedx_chat/static/html/student_view.html
@@ -19,3 +19,9 @@
         </div>
     </div>
 </div>
+<script
+  defer
+  id="ze-snippet"
+  type='application/javascript'
+  src="https://static.zdassets.com/ekr/snippet.js?key=8ef9ef96-3317-40a9-8ef6-de0737503caa"
+></script>