Update app.py

app.py

@@ -2,7 +2,6 @@ import PyPDF2
 import nltk
 from nltk.tokenize import sent_tokenize
 import random
-import requests
 import streamlit as st
 from langchain_openai import OpenAI
 
@@ -10,9 +9,8 @@ from langchain_openai import OpenAI
 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)
@@ -37,13 +35,6 @@ def generate_mcqs_on_topic(text, topic, num_mcqs=5):
     return mcqs
 
 def generate_question_with_chatgpt(context):
-
-    
-    headers = {
-        "Content-Type": "application/json",
-        "Authorization": f"Bearer {OPENAI_API_KEY}",
-    }
-
     # Initializing the default value
     generated_question = {
         'content': "Unable to generate a question..",
@@ -59,7 +50,7 @@ def generate_question_with_chatgpt(context):
         ],
     }
 
-    response = requests.post(CHATGPT_API_ENDPOINT, json=data, headers=headers)
+    response = openai.complete(data)
     result = response.json()
 
     print("API Response:", result)  # Add this line for debugging
@@ -105,6 +96,7 @@ def main():
 if __name__ == "__main__":
     main()
 
+