Update app.py

app.py

@@ -1,18 +1,14 @@
 import PyPDF2
 import nltk
-from nltk.tokenize import sent_tokenize
 import random
-import requests
 import streamlit as st
 from langchain_openai import OpenAI
 
 # Download NLTK data (if not already downloaded)
 nltk.download('punkt')
-nltk.download('averaged_perceptron_tagger')
 
-# ChatGPT API endpoint
-CHATGPT_API_ENDPOINT = "https://api.openai.com/v1/chat/completions"
-OPENAI_API_KEY = "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"
+# LangChain OpenAI wrapper
+openai = OpenAI(api_key="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")
 
 def extract_text_from_pdf(pdf_file):
     pdf_reader = PyPDF2.PdfReader(pdf_file)
@@ -31,19 +27,12 @@ def generate_mcqs_on_topic(text, topic, num_mcqs=5):
     mcqs = []
     for sentence in selected_sentences:
         # Use ChatGPT for interactive question generation
-        chatgpt_question = generate_question_with_chatgpt(sentence)
+        chatgpt_question = generate_question_with_chatgpt(sentence, topic)
         mcqs.append(chatgpt_question)
 
     return mcqs
 
-def generate_question_with_chatgpt(context):
-
-    
-    headers = {
-        "Content-Type": "application/json",
-        "Authorization": f"Bearer {OPENAI_API_KEY}",
-    }
-
+def generate_question_with_chatgpt(context, topic):
     # Initializing the default value
     generated_question = {
         'content': "Unable to generate a question..",
@@ -57,11 +46,11 @@ def generate_question_with_chatgpt(context):
         "max_tokens": 1024,
         "messages": [
             {"role": "system", "content": "You are a helpful assistant."},
-            {"role": "user", "content": f"What is the question for the following? {context}"},
+            {"role": "user", "content": f"What is the question on {topic} for the following? {context}"},
         ],
     }
 
-    response = requests.post(CHATGPT_API_ENDPOINT, json=data, headers=headers)
+    response = openai.chat.completions.create(data)
     result = response.json()
 
     print("API Response:", result)  # Add this line for debugging
@@ -78,7 +67,6 @@ def generate_question_with_chatgpt(context):
 
     return generated_question
 
-
 def main(): 
     # Title of the Application
     st.header("🤖CB Quiz Generator🧠", divider='rainbow')