Update app.py

app.py

@@ -1,141 +1,111 @@
-# app.py
 import gradio as gr
-import requests
 import os
+import requests
 
-# ------------------ GROQ SETTINGS ------------------
-GROQ_API_KEY = os.getenv("GROQ_API_KEY") or "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"
-GROQ_MODEL = "llama3-70b-8192"
+# ========== 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"
 
-def query_gpt(prompt):
+def query_groq_model(prompt):
     headers = {
         "Authorization": f"Bearer {GROQ_API_KEY}",
         "Content-Type": "application/json"
     }
+
     data = {
         "model": GROQ_MODEL,
-        "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"
-        ]
-    },
-    "1st Year": {
-        "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"
-        ]
-    },
-    "2nd Year": {
-        "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"
-        ]
+        "messages": [{"role": "user", "content": prompt}],
+        "temperature": 0.7
     }
+
+    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"]
 }
 
-# ------------------ 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}'."
+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"
+    ]
+}
 
-    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."
+task_options = ["Summarize Important Points", "Generate MCQs", "Simplify Concepts"]
 
-    gpt_response = query_gpt(prompt)
+# ========== APP UI FUNCTION ==========
 
-    # 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})"
+def update_subjects(selected_class):
+    return gr.Dropdown.update(choices=subject_map.get(selected_class, []), value=None)
 
-    return gpt_response
+def update_chapters(selected_subject):
+    return gr.Dropdown.update(choices=chapter_map.get(selected_subject, []), value=None)
 
-# ------------------ 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!")
+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."
 
-    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")
+    prompt = f"""
+You are an AI Textbook Tutor helping a student of Class {selected_class}.
+Subject: {selected_subject}
+Chapter: {selected_chapter}
+Task: {selected_task}
 
-    run_button = gr.Button("Run Tutor 🧠")
-    output_box = gr.Markdown()
+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.
+"""
+
+    ai_output = query_groq_model(prompt)
+
+    # 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(' ', '+')})"
+
+    return f"{ai_output}\n\n📺 **Watch this topic on YouTube:** {youtube_link}"
+
+# ========== GRADIO UI ==========
 
-    def update_subjects(selected_class):
-        return gr.update(choices=list(class_subject_chapters[selected_class].keys()), value=None)
+with gr.Blocks() as demo:
+    gr.Markdown("# 📚 AI Textbook Tutor\nSelect your class, subject, and chapter to get help!")
 
-    def update_chapters(selected_class, selected_subject):
-        return gr.update(choices=class_subject_chapters[selected_class][selected_subject], value=None)
+    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")
 
-    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 = gr.Button("Run Tutor 🧠")
+    output_box = gr.Markdown()
+
+    class_choice.change(fn=update_subjects, inputs=class_choice, outputs=subject_choice)
+    subject_choice.change(fn=update_chapters, inputs=subject_choice, outputs=chapter_choice)
 
     run_button.click(
-        tutor_bot,
-        inputs=[class_dropdown, subject_dropdown, chapter_dropdown, action_dropdown],
+        fn=run_ai_tutor,
+        inputs=[class_choice, subject_choice, chapter_choice, task_choice],
         outputs=output_box
     )
 
-app.launch()
+demo.launch()
+
+
+