Update app.py

app.py

@@ -3,23 +3,26 @@ from youtube_transcript_api import YouTubeTranscriptApi
 import re
 import tempfile
 import os
-import whisper
 import warnings
 from groq import Groq
+from whisper_jax import FlaxWhisperPipline
 
 # Suppress specific warning
 warnings.filterwarnings("ignore", message="FP16 is not supported on CPU; using FP32 instead")
 
 # Set up Groq client
-client = Groq(
-    api_key=os.environ.get("GROQ_API_KEY"),
-)
+client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-# Function to transcribe audio
+# Instantiate Whisper JAX pipeline
+pipeline = FlaxWhisperPipline("openai/whisper-large-v2")
+
+# Function to transcribe audio using Whisper JAX
 def transcribe_audio(file_path):
-    model = whisper.load_model("base")
-    result = model.transcribe(file_path)
-    return result["text"]
+    # JIT compile the forward call - slow, but we only do once
+    text = pipeline(file_path)
+    # Used cached function thereafter - super fast!!
+    text = pipeline(file_path)
+    return text
 
 # Function to get transcript from YouTube
 def get_transcript(url):
@@ -27,7 +30,6 @@ def get_transcript(url):
         video_id_match = re.search(r"(?:v=|\/)([0-9A-Za-z_-]{11}).*", url)
         if not video_id_match:
             return "Error: Invalid YouTube URL"
-
         video_id = video_id_match.group(1)
         transcript = YouTubeTranscriptApi.get_transcript(video_id)
         transcript_text = ' '.join([entry['text'] for entry in transcript])
@@ -220,15 +222,11 @@ if st.session_state.generated_quiz:
 
     if st.button("Submit Answers"):
         if "questions" in st.session_state and st.session_state.questions:
-            with st.spinner('Processing your answers...'):
-                feedback = check_answers(st.session_state.questions, st.session_state.user_answers)
-                st.write("## Feedback")
-                for i, item in enumerate(feedback):
-                    with st.expander(f"Question {i+1} Feedback"):
-                        st.write(f"### {item['question']}")
-                        st.write(f"**Your answer:** {item['user_answer']}")
-                        st.write(f"**Correct answer:** {item['correct_answer']}")
-                        if item['status'] == "Incorrect":
-                            st.write(f"**Explanation:** {item['explanation']}")
-        else:
-            st.write("Please generate the quiz first.")
+            with st.spinner('ProcessingIf `whisper_jax` is not found, it might not be available on PyPI or installed properly. If you need to use an alternative or similar library, ensure it's correctly installed. Let's try another approach using `whisper` from OpenAI, which should be available on PyPI and can be used similarly.
+
+### Updated `requirements.txt`
+```txt
+streamlit
+youtube_transcript_api
+groq
+whisper