Update app.py

app.py

@@ -1,111 +1,144 @@
 import gradio as gr
-import os
 import requests
+import os
 
-# ========== GROQ API SETUP ==========
-GROQ_API_KEY = "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"
-GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
-GROQ_MODEL = "mixtral-8x7b-32768"
+# ------------------ GROQ SETTINGS ------------------
+GROQ_API_KEY = os.getenv("GROQ_API_KEY") or "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"
+GROQ_MODEL = "llama3-70b-8192"
 
-def query_groq_model(prompt):
+def query_gpt(prompt):
     headers = {
         "Authorization": f"Bearer {GROQ_API_KEY}",
         "Content-Type": "application/json"
     }
-
     data = {
         "model": GROQ_MODEL,
-        "messages": [{"role": "user", "content": prompt}],
-        "temperature": 0.7
+        "messages": [{"role": "user", "content": prompt}]
+    }
+    response = requests.post("https://api.groq.com/openai/v1/chat/completions", json=data, headers=headers)
+    return response.json()['choices'][0]['message']['content']
+
+# ------------------ SYLLABUS DATA ------------------
+class_subject_chapters = {
+    "9": {
+        "Physics": [
+            "Introduction to Physics", "Kinematics", "Dynamics", "Turning Effect of Forces",
+            "Gravitation", "Work and Energy", "Properties of Matter", "Thermal Properties of Matter",
+            "Transfer of Heat"
+        ],
+        "Biology": [
+            "Introduction to Biology", "Solving a Biological Problem", "Biodiversity",
+            "Cells and Tissues", "Cell Cycle", "Enzymes", "Bioenergetics", "Nutrition",
+            "Transport"
+        ],
+        "Chemistry": [
+            "Fundamentals of Chemistry", "Structure of Atoms", "Periodic Table",
+            "Structure of Molecules", "Physical States of Matter", "Solutions",
+            "Electrochemistry", "Chemical Reactivity"
+        ]
+    },
+    "10": {
+        "Physics": [
+            "Simple Harmonic Motion and Waves", "Sound", "Geometrical Optics",
+            "Electrostatics", "Current Electricity", "Electromagnetism",
+            "Basic Electronics", "Information and Communication Technology",
+            "Atomic and Nuclear Physics"
+        ],
+        "Biology": [
+            "Gaseous Exchange", "Homeostasis", "Coordination and Control",
+            "Support and Movement", "Reproduction", "Inheritance",
+            "Man and His Environment", "Biotechnology", "Pharmacology"
+        ],
+        "Chemistry": [
+            "Chemical Equilibrium", "Acids, Bases and Salts", "Organic Chemistry",
+            "Hydrocarbons", "Biochemistry", "Environmental Chemistry", "Chemical Industries"
+        ]
+    },
+    "11": {
+        "Physics": [
+            "Measurements", "Vectors and Equilibrium", "Motion and Force",
+            "Work and Energy", "Circular Motion", "Fluid Dynamics",
+            "Oscillations", "Waves", "Physical Optics", "Thermodynamics"
+        ],
+        "Biology": [
+            "Introduction to Biology", "Biological Molecules", "Enzymes",
+            "The Cell", "Cell Cycle", "Variety of Life", "Kingdom Prokaryotae",
+            "Kingdom Protista", "Kingdom Fungi", "Kingdom Plantae", "Kingdom Animalia"
+        ],
+        "Chemistry": [
+            "Basic Concepts", "Experimental Techniques", "Gases", "Liquids and Solids",
+            "Atomic Structure", "Chemical Bonding", "Thermochemistry",
+            "Chemical Equilibrium", "Ionic Equilibrium", "Redox Reactions"
+        ]
+    },
+    "12": {
+        "Physics": [
+            "Electrostatics", "Current Electricity", "Electromagnetism", "Electromagnetic Induction",
+            "Alternating Current", "Physics of Solids", "Electronics", "Dawn of Modern Physics",
+            "Atomic Spectra", "Nuclear Physics"
+        ],
+        "Biology": [
+            "Homeostasis", "Support and Movement", "Coordination and Control",
+            "Reproduction", "Growth and Development", "Chromosomes and DNA",
+            "Biotechnology", "Evolution", "Ecosystem", "Some Major Ecosystems"
+        ],
+        "Chemistry": [
+            "Periodic Classification", "Transition Elements", "Chemical Reactivity",
+            "Alcohols and Phenols", "Aldehydes and Ketones", "Carboxylic Acids",
+            "Amines", "Polymers", "Biochemistry", "Environmental Chemistry"
+        ]
     }
-
-    response = requests.post(GROQ_API_URL, headers=headers, json=data)
-
-    if response.status_code == 200:
-        return response.json()["choices"][0]["message"]["content"]
-    else:
-        return f"❌ Error: {response.status_code} - {response.text}"
-
-# ========== CLASS, SUBJECT, CHAPTER OPTIONS ==========
-
-subject_map = {
-    "9th": ["Physics", "Chemistry", "Biology"],
-    "10th": ["Physics", "Chemistry", "Biology"],
-    "1st Year": ["Physics", "Chemistry", "Biology"],
-    "2nd Year": ["Physics", "Chemistry", "Biology"]
-}
-
-chapter_map = {
-    "Physics": [
-        "Physical Quantities and Units", "Kinematics", "Dynamics", "Turning Effect of Forces",
-        "Gravitation", "Work and Energy", "Properties of Matter", "Thermal Properties of Matter",
-        "Transfer of Heat", "Simple Harmonic Motion", "Sound", "Geometrical Optics",
-        "Electrostatics", "Current Electricity", "Electromagnetism", "Electronics", "Dawn of Modern Physics"
-    ],
-    "Chemistry": [
-        "Fundamentals of Chemistry", "Structure of Atom", "Periodic Table", "Chemical Bonding",
-        "States of Matter", "Solutions", "Electrochemistry", "Chemical Kinetics", "Organic Chemistry", "Environmental Chemistry"
-    ],
-    "Biology": [
-        "Introduction to Biology", "Biological Molecules", "Cell Biology", "Enzymes",
-        "Bioenergetics", "Nutrition", "Transport", "Respiration", "Coordination", "Homeostasis"
-    ]
 }
 
-task_options = ["Summarize Important Points", "Generate MCQs", "Simplify Concepts"]
-
-# ========== APP UI FUNCTION ==========
-
-def update_subjects(selected_class):
-    return gr.Dropdown.update(choices=subject_map.get(selected_class, []), value=None)
-
-def update_chapters(selected_subject):
-    return gr.Dropdown.update(choices=chapter_map.get(selected_subject, []), value=None)
+# ------------------ MAIN BOT FUNCTION ------------------
+def tutor_bot(selected_class, selected_subject, selected_chapter, action_type):
+    prompt = f"The user is studying in Class {selected_class}. Give a detailed response for the chapter '{selected_chapter}' in the subject '{selected_subject}'."
 
-def run_ai_tutor(selected_class, selected_subject, selected_chapter, selected_task):
-    if not all([selected_class, selected_subject, selected_chapter, selected_task]):
-        return "⚠️ Please complete all selections."
+    if action_type == "Summarize Important Points":
+        prompt = f"Summarize the key points of the chapter '{selected_chapter}' from Class {selected_class} {selected_subject}."
+    elif action_type == "Generate MCQs":
+        prompt = f"Create 5 multiple choice questions (MCQs) with 4 options each for the chapter '{selected_chapter}' from Class {selected_class} {selected_subject}."
+    elif action_type == "Simplify Concepts":
+        prompt = f"Explain in very simple terms the important concepts from the chapter '{selected_chapter}' from Class {selected_class} {selected_subject} for a 14-year-old student."
 
-    prompt = f"""
-You are an AI Textbook Tutor helping a student of Class {selected_class}.
-Subject: {selected_subject}
-Chapter: {selected_chapter}
-Task: {selected_task}
+    gpt_response = query_gpt(prompt)
 
-Generate a student-friendly response. If the task is to generate MCQs, include 3 MCQs with options and correct answers. Keep it concise and clear.
-"""
+    # Add YouTube search link
+    search_query = f"{selected_class} {selected_subject} {selected_chapter} in Urdu"
+    youtube_link = f"https://www.youtube.com/results?search_query={search_query.replace(' ', '+')}"
+    gpt_response += f"\n\n🎥 **Watch this topic on YouTube:** [Click here]({youtube_link})"
 
-    ai_output = query_groq_model(prompt)
+    return gpt_response
 
-    # Create YouTube search link
-    youtube_query = f"{selected_class}+{selected_subject}+{selected_chapter}+in+Urdu"
-    youtube_link = f"[Click here](https://www.youtube.com/results?search_query={youtube_query.replace(' ', '+')})"
+# ------------------ GRADIO UI ------------------
+with gr.Blocks(title="AI Textbook Tutor") as app:
+    gr.Markdown("## 📘 AI Textbook Tutor\nSelect your class, subject, and chapter to get summaries, MCQs, or simplified explanations!")
 
-    return f"{ai_output}\n\n📺 **Watch this topic on YouTube:** {youtube_link}"
+    class_dropdown = gr.Dropdown(label="Select Class", choices=list(class_subject_chapters.keys()))
+    subject_dropdown = gr.Dropdown(label="Select Subject")
+    chapter_dropdown = gr.Dropdown(label="Select Chapter")
+    action_dropdown = gr.Radio(["Summarize Important Points", "Generate MCQs", "Simplify Concepts"], label="Select Task")
 
-# ========== GRADIO UI ==========
-
-with gr.Blocks() as demo:
-    gr.Markdown("# 📚 AI Textbook Tutor\nSelect your class, subject, and chapter to get help!")
+    run_button = gr.Button("Run Tutor 🧠")
+    output_box = gr.Markdown()  # Changed from Textbox to Markdown for clickable links
 
-    class_choice = gr.Dropdown(label="Select Class", choices=list(subject_map.keys()))
-    subject_choice = gr.Dropdown(label="Select Subject")
-    chapter_choice = gr.Dropdown(label="Select Chapter")
-    task_choice = gr.Radio(task_options, label="Select Task")
+    def update_subjects(selected_class):
+        return gr.update(choices=list(class_subject_chapters[selected_class].keys()), value=None)
 
-    run_button = gr.Button("Run Tutor 🧠")
-    output_box = gr.Markdown()
+    def update_chapters(selected_class, selected_subject):
+        return gr.update(choices=class_subject_chapters[selected_class][selected_subject], value=None)
 
-    class_choice.change(fn=update_subjects, inputs=class_choice, outputs=subject_choice)
-    subject_choice.change(fn=update_chapters, inputs=subject_choice, outputs=chapter_choice)
+    class_dropdown.change(update_subjects, inputs=class_dropdown, outputs=subject_dropdown)
+    subject_dropdown.change(update_chapters, inputs=[class_dropdown, subject_dropdown], outputs=chapter_dropdown)
 
     run_button.click(
-        fn=run_ai_tutor,
-        inputs=[class_choice, subject_choice, chapter_choice, task_choice],
+        tutor_bot,
+        inputs=[class_dropdown, subject_dropdown, chapter_dropdown, action_dropdown],
         outputs=output_box
     )
 
-demo.launch()
+app.launch()
+