Update app.py

app.py

@@ -1,24 +1,21 @@
 import os
-import tempfile
 from datetime import datetime
 from PyPDF2 import PdfReader
 from docx import Document
 import streamlit as st
-from transformers import pipeline
+from groq import Groq
 
-# ✅ Page config MUST be first Streamlit command
+# ✅ Streamlit Page Config
 st.set_page_config(page_title="AI Study Plan Assistant", layout="wide")
 
-# Load models with caching
+# ✅ Load Groq Client
 @st.cache_resource
-def load_models():
-    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-    qa_pipeline = pipeline("question-answering", model="distilbert-base-uncased-distilled-squad")
-    return summarizer, qa_pipeline
+def load_groq_client():
+    return Groq(api_key=os.getenv("GROQ_API_KEY"))
 
-summarizer, qa_pipeline = load_models()
+groq_client = load_groq_client()
 
-# Extract text from supported formats
+# ✅ Extract text from files
 def extract_text(file):
     ext = os.path.splitext(file.name)[1].lower()
     if ext == ".txt":
@@ -30,12 +27,25 @@ def extract_text(file):
         pdf_reader = PdfReader(file)
         text = ""
         for page in pdf_reader.pages:
-            text += page.extract_text() + "\n"
+            if page.extract_text():
+                text += page.extract_text() + "\n"
         return text
     else:
         raise ValueError("Only .txt, .docx, and .pdf files are supported.")
 
-# Generate study plan
+# ✅ Query Groq LLM
+def query_groq(prompt, temperature=0.5):
+    try:
+        chat = groq_client.chat.completions.create(
+            messages=[{"role": "user", "content": prompt}],
+            model="mixtral-8x7b-32768",
+            temperature=temperature
+        )
+        return chat.choices[0].message.content.strip()
+    except Exception as e:
+        return f"⚠️ Groq API Error: {str(e)}"
+
+# ✅ Study Plan Generation
 def generate_plan(file, hours_per_day, exam_date, language):
     try:
         content = extract_text(file)
@@ -46,26 +56,37 @@ def generate_plan(file, hours_per_day, exam_date, language):
         if days <= 0:
             return "❌ Exam date must be in the future."
 
-        prompt = f"Create a {days}-day study plan for this syllabus:\n{content}\nDaily study time: {hours_per_day} hours."
-        if language == "Urdu":
-            prompt += "\nTranslate the plan into Urdu."
-
-        summary = summarizer(prompt, max_length=512, min_length=150, do_sample=False)
-        return summary[0]['summary_text']
+        prompt = f"""Create a detailed {days}-day study plan from the syllabus below.
+You should assume the user can study {hours_per_day} hours per day.
+Output should be in {language}.
 
+Syllabus:
+\"\"\"
+{content}
+\"\"\"
+"""
+        return query_groq(prompt)
     except Exception as e:
         return f"⚠️ Error: {str(e)}"
 
-# Ask a question
+# ✅ Question Answering
 def ask_question(file, question):
     try:
         context = extract_text(file)
-        result = qa_pipeline({"context": context, "question": question})
-        return result['answer']
+        prompt = f"""Based on the following study material, answer the question below:
+
+Study Material:
+\"\"\"
+{context}
+\"\"\"
+
+Question: {question}
+Answer:"""
+        return query_groq(prompt)
     except Exception as e:
         return f"⚠️ Error: {str(e)}"
 
-# Streamlit UI
+# ✅ Streamlit UI
 st.sidebar.title("📚 Study Assistant Options")
 uploaded_file = st.sidebar.file_uploader("Upload syllabus (.txt, .docx, .pdf)", type=["txt", "docx", "pdf"])
 study_hours = st.sidebar.number_input("Study hours per day", min_value=1, max_value=12, value=3)