Update app.py

app.py

@@ -1,6 +1,5 @@
 # Step 2: Import libraries
 import gradio as gr
-from transformers import pipeline
 
 # Step 3: Define College Data
 college_about = """
@@ -10,276 +9,130 @@ The college started its classes in the afternoon at Government High School Daula
 With the help of the EDO Education, the possession of the Government Girls High School (Old) Daulatpur was given to the college.
 """
 
-faculty_data = """
-1. Prof Mahfooz Khan (Associate Prof, Islamiat)
-2. Muhammad Aslam Palli (Associate Prof, Physics)
-3. Abdul Rahman Gaincho (Botany)
-4. M. Essa Samego (Lecturer, Islamiat)
-5. Ali Aijaz Dahri (Lecturer, Chemistry)
-6. Asif Ali Rajput (Lecturer, Physics)
-7. Fida Hussain (Lecturer, Math)
-8. Zulfiqar Ali (Senior Librarian)
-"""
-
-facilities = """
-Facilities available:
-1. Digital library with 20+ computers
-2. Parks: Alquran Park and Botany Park with 50+ plants
-3. Labs: Zoology, Botany, Physics, Chemistry
-4. Parking area
-5. Solar system and RO Plant
-"""
-
-# Step 4: Define College Dataset (from your data dictionary)
-# Define the dataset
+# Restructured Data
 data = {
-    "Questions": [
-        "tell me name of college?",
-        "tell me establish of college?",
-        "who is principal of college?",
-        "what a programs in college?",
-        "what a programs in college?",
-        "what a programs in college?",
-        "what a programs in college?",
-        "what a programs in college?",
-        "where is located?",
-        "islamiat teacher?",
-        "physics teacher?",
-        "botany teacher?",
-        "islamiat teacher?",
-        "chemistry teacher?",
-        "physics teacher?",
-        "math teacher?",
-        "who is librarian",
-        "who is DPE",
-        "lab assistant?",
-        "clerk?",
-        "clerk?",
-        "clerk?",
-        "lab assistant?",
-        "lab assistant?",
-        "lab assistant?",
-        "who is sanitary worker?",
-        "facilities?",
-        "facilities?",
-        "facilities?",
-        "facilities?",
-        "facilities?",
-        "facilities?",
-        "facilities?",
-        "facilities?",
-        "facilities?",
-        "facilities?",
-        "facilities?",
+    "Programs": [
+        "ADSE-I", "ADSE-II", "ADAA-I", "ADAA-II", "BS (Zoology) (Proposed for this year)"
     ],
-    "Category": [
-        "College Name",
-        "Established",
-        "Principal",
-        "Programs",
-        "Programs",
-        "Programs",
-        "Programs",
-        "Programs",
-        "Location",
-        "Teaching Staff",
-        "Teaching Staff",
-        "Teaching Staff",
-        "Teaching Staff",
-        "Teaching Staff",
-        "Teaching Staff",
-        "Teaching Staff",
-        "Teaching Staff",
-        "Teaching Staff",
-        "Non-Teaching Staff",
-        "Non-Teaching Staff",
-        "Non-Teaching Staff",
-        "Non-Teaching Staff",
-        "Non-Teaching Staff",
-        "Non-Teaching Staff",
-        "Non-Teaching Staff",
-        "Non-Teaching Staff",
-        "Facilities",
-        "Facilities",
-        "Facilities",
-        "Facilities",
-        "Facilities",
-        "Facilities",
-        "Facilities",
-        "Facilities",
-        "Facilities",
-        "Facilities",
-        "Facilities",
-    ],
-    "Name": [
-        "Government Boys Degree College Daulatpur",
-        "2003",
-        "Prof Irshad Ali Otho",
-        "ADSE-I",
-        "ADSE-II",
-        "ADAA-",
-        "ADAA",
-        "(Zoology),Proposed this year,",
-        "Bypass Daulatpur",
-        "Prof Mahfooz Khan",
-        "Muhammad Aslam Palli",
-        "Abdul Rahman Gaincho",
-        "M. Essa Samego",
-        "Ali Aijaz Dahri",
-        "Asif Ali Rajput",
-        "Fida Hussain",
-        "Zulfiqar Ali",
-        "Naseer Ahmed",
-        "Mr Abear Lund",
-        "Mr Younis Unar",
-        "Mr Muhammad Bughio",
-        "Mr Dadan Khan,",
-        "Mr Rahib Ali Unar",
-        "Mr M Saleem Unar",
-        "Mr Shahbaz Khan",
-        "Mr Peruaz Ali",
-        "Digital Library",
-        "Parks",
-        "Building",
-        "Zoology Lab",
-        "Botany Lab",
-        "Physics Lab",
-        "Chemistry Lab",
-        "Parking Area",
-        "Solar System",
+    "Teachers": {
+        "Islamiat": ["Prof Mahfooz Khan (Associate Prof)", "M. Essa Samego (Lecturer)"],
+        "Physics": ["Muhammad Aslam Palli (Associate Prof)", "Asif Ali Rajput (Lecturer)"],
+        "Botany": ["Abdul Rahman Gaincho (Lecturer)"],
+        "Chemistry": ["Ali Aijaz Dahri (Lecturer)"],
+        "Math": ["Fida Hussain (Lecturer)"],
+        "Librarian": ["Zulfiqar Ali (Senior Librarian)"],
+        "DPE": ["Naseer Ahmed (DPE)"],
+    },
+    "Facilities": [
+        "Digital library with 20+ computers",
+        "Parks: Alquran Park and Botany Park with 50+ plants",
+        "Building: Two-floor building with 24 rooms",
+        "Labs: Zoology, Botany, Physics, Chemistry",
+        "Parking area",
+        "Solar system",
         "RO Plant",
     ],
-    "Designation": [
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "Associate Professor",
-        "Associate Professor",
-        "Lecturer",
-        "Lecturer",
-        "Lecturer",
-        "Lecturer",
-        "Lecturer",
-        "Senior Librarian",
-        "DPE",
-        "Assistant",
-        "Senior Clerk",
-        "Senior Clerk",
-        "Junior Clerk",
-        "Lab Assistant",
-        "Lab Attendant",
-        "Lab Attendant",
-        "Sanitary Worker",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-    ],
-    "Subject/Details": [
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "Islamiat",
-        "Physics",
-        "Botany",
-        "Islamiat",
-        "Chemistry",
-        "Physics",
-        "Math",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "NA",
-        "20+ Computers",
-        "Alquran Park and Botany Park with 50+ plats",
-        "Two-floor building with 24 rooms",
-        "Available",
-        "Available",
-        "Available",
-        "Available",
-        "Available",
-        "Available",
-        "Available",
-    ],
+    "Location": "Bypass Daulatpur",
+    "Principal": "Prof Irshad Ali Otho",
+    "Established": "2003",
+    "Contact": {
+        "Location": "National Highway N6 bypass Daulatpur",
+        "Phone": "+92 300 3003249",
+        "Email": "[email protected]",
+        "Facebook": "https://www.facebook.com/share/19j9Z1iEvz/"
+    }
 }
 
-# Print the data (optional)
-#print(data)
+# Helper function to answer questions
+def get_answer(question):
+    # Convert question to lowercase for easier matching
+    question = question.lower()
 
+    # Check for general categories
+    if "program" in question or "course" in question:
+        return f"Programs offered:\n- " + "\n- ".join(data["Programs"])
 
-# Step 5: Define a Question-Answering Model (Hugging Face)
-qa_model = pipeline("question-answering", model="deepset/roberta-base-squad2")
+    if "teacher" in question:
+        if "islamiat" in question:
+            return f"Islamiat Teachers:\n- " + "\n- ".join(data["Teachers"]["Islamiat"])
+        elif "physics" in question:
+            return f"Physics Teachers:\n- " + "\n- ".join(data["Teachers"]["Physics"])
+        elif "botany" in question:
+            return f"Botany Teacher:\n- " + "\n- ".join(data["Teachers"]["Botany"])
+        elif "chemistry" in question:
+            return f"Chemistry Teacher:\n- " + "\n- ".join(data["Teachers"]["Chemistry"])
+        elif "math" in question:
+            return f"Math Teacher:\n- " + "\n- ".join(data["Teachers"]["Math"])
+        elif "librarian" in question:
+            return f"Librarian:\n- " + "\n- ".join(data["Teachers"]["Librarian"])
+        elif "dpe" in question:
+            return f"DPE:\n- " + "\n- ".join(data["Teachers"]["DPE"])
+        else:
+            # Return all teachers if no specific subject is mentioned
+            all_teachers = "\n".join(
+                [f"{subject}: " + ", ".join(teachers) for subject, teachers in data["Teachers"].items()]
+            )
+            return f"All Teachers:\n{all_teachers}"
 
-        # Helper function for answering dataset questions
-def get_predefined_answer(question):
-    for q, name in zip(data["Questions"], data["Name"]):  # Use "Name" for predefined answers
-        if question.lower() in q.lower():
-            return name
-    return "Sorry, I don't have an answer for that. Please ask a different question."
+    if "facility" in question:
+        return f"Facilities available:\n- " + "\n- ".join(data["Facilities"])
 
-# Define answer query function with dataset context
-def answer_query(question):
-    predefined_answer = get_predefined_answer(question)
-    if predefined_answer != "Sorry, I don't have an answer for that. Please ask a different question.":
-        return predefined_answer
-    else:
-        # Fallback to model-based answer if question is not predefined
-        context = f"{college_about}\n\n{faculty_data}\n\n{facilities}"
-        response = qa_model(question=question, context=context)
-        return response['answer']
+    if "location" in question:
+        return f"College Location: {data['Location']}"
 
+    if "principal" in question:
+        return f"Principal: {data['Principal']}"
 
-# Step 6: Create Gradio Interface
+    if "establish" in question or "founded" in question:
+        return f"Established in: {data['Established']}"
+
+    if "contact" in question:
+        contact_info = data["Contact"]
+        return (
+            f"Contact Details:\n"
+            f"- Location: {contact_info['Location']}\n"
+            f"- Phone: {contact_info['Phone']}\n"
+            f"- Email: {contact_info['Email']}\n"
+            f"- Facebook: {contact_info['Facebook']}"
+        )
+
+    return "Sorry, I don't have an answer for that. Please ask something else."
+
+
+# Step 4: Create Gradio Interface
 with gr.Blocks(theme=gr.themes.Monochrome()) as app:
     gr.Markdown("## Government Boys Degree College Daulatpur")
-    gr.Markdown("### Welcome to our College Information RAG App")
-    gr.Image("college_logo.png", label="College Logo")  # Upload your logo as college_logo.png
+    gr.Markdown("### Welcome to our College Information App")
+    gr.Image("college_logo.png", label="College Logo")  # Ensure "college_logo.png" exists
+
 
     with gr.Tabs():
         with gr.Tab("About College"):
             gr.Markdown(college_about)
-        with gr.Tab("Faculty and Staff"):
-            gr.Markdown(faculty_data)
+        with gr.Tab("Programs"):
+            gr.Markdown("\n".join(data["Programs"]))
+        with gr.Tab("Teachers"):
+            all_teachers = "\n".join(
+                [f"{subject}: " + ", ".join(teachers) for subject, teachers in data["Teachers"].items()]
+            )
+            gr.Markdown(all_teachers)
         with gr.Tab("Facilities"):
-            gr.Markdown(facilities)
+            gr.Markdown("\n".join(data["Facilities"]))
         with gr.Tab("Ask a Question"):
-            question = gr.Textbox(label="Ask about the college")
+            question = gr.Textbox(label="Ask a question about the college")
             answer = gr.Textbox(label="Answer")
-            question.submit(answer_query, inputs=question, outputs=answer)
-        with gr.Tab("Predefined Data"):
-            predefined_data = "\n".join([f"{q} - Category: {cat}, Name: {name}, Designation: {desig}, Details: {details}"
-                             for q, cat, name, desig, details in zip(data["Questions"], data["Category"], data["Name"], data["Designation"], data["Subject/Details"])])
-            gr.Markdown(predefined_data)
+            question.submit(get_answer, inputs=question, outputs=answer)
         with gr.Tab("Contact Us"):
-            gr.Markdown("**Contact Details:**\n- Location: National Highway N6 bypass Daulatpur\n- Phone: +92 300 3003249\n- Email: [email protected]")
-            gr.Markdown("[Visit our Facebook Page](https://www.facebook.com/share/19j9Z1iEvz/)")
+            contact_info = data["Contact"]
+            gr.Markdown(
+                f"**Contact Details:**\n"
+                f"- Location: {contact_info['Location']}\n"
+                f"- Phone: {contact_info['Phone']}\n"
+                f"- Email: {contact_info['Email']}\n"
+                f"[Visit our Facebook Page]({contact_info['Facebook']})"
+            )
 
     gr.Markdown("### Made by Musawir Manzoor")
 
-# Step 7: Launch the App
+# Step 5: Launch the App
 app.launch()
-