Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 from transformers import pipeline
 import PyPDF2
-import json
+import re
 
-# 📌 Step 1: Extract text from PDF
+# Load syllabus from PDF
 def read_pdf(file_path):
     try:
         with open(file_path, "rb") as file:
@@ -15,14 +15,14 @@ def read_pdf(file_path):
 
 syllabus_text = read_pdf("Syllabus.pdf")
 
-# 📌 Step 2: Extract subjects and topics
+# Extract subjects and topics
 def extract_subjects_and_topics(text):
     subjects = {}
     current_subject = None
 
     for line in text.split("\n"):
         line = line.strip()
-        if line.isupper():  # Assuming subject names are in uppercase
+        if re.match(r"^[A-Z ]+$", line):  # Matches UPPERCASE subject names
             current_subject = line
             subjects[current_subject] = []
         elif current_subject and line:
@@ -32,30 +32,28 @@ def extract_subjects_and_topics(text):
 
 subjects_data = extract_subjects_and_topics(syllabus_text)
 
-# 📌 Step 3: Convert to JSON format for easy searching
-subjects_json = json.dumps(subjects_data, indent=4)
-
-# 📌 Load AI Model for Chatbot
+# Load AI Model for Chatbot
 chatbot = pipeline("text-generation", model="facebook/blenderbot-400M-distill")
 
-# 📌 Step 4: Chat Function
+# Chat function
 def chat_response(message):
     message = message.lower()
 
-    # If user asks for subjects
     if "subjects" in message:
         return "📚 Available Subjects:\n\n" + "\n".join(subjects_data.keys())
 
-    # If user asks for topics under a subject
     for subject, topics in subjects_data.items():
         if subject.lower() in message:
             return f"📖 Topics under {subject}:\n\n" + "\n".join(topics)
 
-    # If chatbot response is needed
-    response = chatbot(message, max_length=100, do_sample=True)
-    return response[0]['generated_text']
+    for subject, topics in subjects_data.items():
+        for topic in topics:
+            if topic.lower() in message:
+                return f"📌 {topic} is covered under {subject}. Refer to your syllabus for details."
+
+    return "❌ Topic not found in syllabus. Please check the spelling or ask about a different topic."
 
-# 📌 Step 5: Create Gradio Interface
+# Create Gradio Interface
 iface = gr.Interface(
     fn=chat_response,
     inputs="text",
@@ -64,6 +62,6 @@ iface = gr.Interface(
     description="Ask me about syllabus subjects, topics, or general questions!"
 )
 
-# 📌 Step 6: Launch App
+# Launch App
 if __name__ == "__main__":
     iface.launch()