Update file_utils.py

file_utils.py

@@ -1,5 +1,9 @@
 import fitz  # PyMuPDF
 from pptx import Presentation
+from openai import OpenAI
+import os
+
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 
 def extract_text_from_pdf(pdf_path):
@@ -45,3 +49,71 @@ def extract_text_from_file(file):
         return extract_text_from_pptx(file.name)
     else:
         return "Unsupported file type."
+
+
+def generate_summary(text):
+    prompt = f"""Please summarize the following study material in a concise and organized manner:
+
+{text}
+"""
+    response = client.chat.completions.create(
+        model="gpt-4",
+        messages=[
+            {"role": "system", "content": "You are a helpful study assistant."},
+            {"role": "user", "content": prompt}
+        ]
+    )
+    return response.choices[0].message.content
+
+
+def generate_flashcards(text):
+    prompt = f"""Based on the content below, create a set of helpful flashcards.
+Each flashcard should be formatted as:
+Q: Question?
+A: Answer.
+
+{text}
+"""
+    response = client.chat.completions.create(
+        model="gpt-4",
+        messages=[
+            {"role": "system", "content": "You are a flashcard generator assistant."},
+            {"role": "user", "content": prompt}
+        ]
+    )
+    return response.choices[0].message.content
+
+
+def generate_quiz(text):
+    prompt = f"""Create a short quiz (3-5 questions) based on the content below.
+Include a mix of multiple choice and short answer questions.
+
+{text}
+"""
+    response = client.chat.completions.create(
+        model="gpt-4",
+        messages=[
+            {"role": "system", "content": "You are a quiz generator assistant."},
+            {"role": "user", "content": prompt}
+        ]
+    )
+    return response.choices[0].message.content
+
+
+def answer_question(text, question):
+    prompt = f"""You are an AI tutor. Answer the following question based only on the content below.
+
+Content:
+{text}
+
+Question:
+{question}
+"""
+    response = client.chat.completions.create(
+        model="gpt-4",
+        messages=[
+            {"role": "system", "content": "You are a helpful and accurate tutor that stays grounded in provided content."},
+            {"role": "user", "content": prompt}
+        ]
+    )
+    return response.choices[0].message.content