Create app.py

app.py

@@ -0,0 +1,68 @@
+import streamlit as st
+import pytube
+from youtube_transcript_api import YouTubeTranscriptApi as yt
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+import os
+from langchain import PromptTemplate
+from langchain import LLMChain
+from langchain_together import Together
+import re
+import json
+
+def Summary_BART(text):
+    checkpoint = "sshleifer/distilbart-cnn-12-6"
+    tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+    model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
+    inputs = tokenizer(text,
+                    max_length=1024,
+                    truncation=True,
+                    return_tensors="pt")
+    summary_ids = model.generate(inputs["input_ids"])
+    summary = tokenizer.batch_decode(summary_ids,
+                                  skip_special_tokens=True,
+                                  clean_up_tokenization_spaces=False)
+    return summary[0]
+
+def YtToQuizz(link,difficulty_level):
+  video_id=pytube.extract.video_id(link)
+  transcript=yt.get_transcript(video_id)
+  data=""
+  for text in transcript:
+    data+=text.get('text')
+  summary=Summary_BART(data)
+  print(summary)
+  mcq_template = """
+    Give a 10 different multiple-choice question MCQ related to the summary: {summary}
+    The difficulty level of the question should be: {difficulty_level}
+
+    Please provide the following in:
+    1. Question
+    2. Correct answer
+    3. Three plausible incorrect answer options
+    4. Proper Mcqs format
+    """ 
+  prompt=PromptTemplate(
+    input_variables=['summary','difficulty_level'],
+    template=mcq_template
+  )
+  llama3 = Together(model = "meta-llama/Llama-3-70b-chat-hf",
+                  max_tokens = 2500
+                  )
+  Generated_mcqs=LLMChain(llm=llama3,prompt=prompt)
+
+  response = Generated_mcqs.invoke({
+    "summary": summary,
+    "difficulty_level": difficulty_level
+  })
+  st.write("MCQ's are following")
+  st.write(response['text'])
+
+
+def main():    
+    st.title("YouTube video Subtitle to MCQ's Quizz")
+    url_link=st.text_area("Enter YouTube video link")
+    diffculity_level=st.selectbox("Select diffculity level:",["Eassy","Medium","Hard"])
+    if st.button("Generate MCQS Quizz"):
+      YtToQuizz(url_link,diffculity_level)
+if __name__ == '__main__':
+    main()