Update app.py

app.py

@@ -1,64 +1,224 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
+from transformers import pipeline
 
-"""
-For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
-"""
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
+# Step 1: Load the Question Answering Pipeline
+qa_pipeline = pipeline("question-answering", model="distilbert-base-uncased-distilled-squad")
 
+# Step 2: Define the College Data
+college_data = {
+    "name": "Government Boys Degree College Daulatpur",
+    "about": """
+    The college was notified to work from 16th July 2003 on the directions of the then care-taker Minister for Education, Mr. Khan Muhammad Dahri.
+    At the beginning, the college had no staff of its own, so it was borrowed from Government Degree College Sakrand.
+    The college started its classes in the afternoon at Government High School Daulatpur as it had no specific building.
+    With the help of the EDO Education, the possession of the Government Girls High School (Old) Daulatpur was given to the college.
+    """,
+    "extended_about": """
+    Government Boys Degree College Daulatpur is one of the leading educational institutions in the region. It aims to provide high-quality education
+    and a conducive environment for academic excellence. The college offers various undergraduate programs and is committed to the intellectual 
+    and professional growth of its students. It strives to nurture responsible citizens who contribute positively to society.
+    """,
+    "programs": ["ADSE-I", "ADSE-II", "ADAA-I", "ADAA-II", "BS (Zoology) (Proposed for this year)"],
+    "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)"],
+    },
+    "staff": {
+        "clerks": ["Mr. Younis Unar (Senior Clerk)", "Mr. Muhammad Bughio (Senior Clerk)", "Mr. Dadan Khan (Junior Clerk)"],
+        "lab_assistants": ["Mr. Rahib Ali Unar (Lab Assistant)", "Mr. Shahbaz Khan (Lab Attendant)"],
+        "other": ["Mr. Peruaz Ali (Sanitary Worker)"],
+    },
+    "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",
+    ],
+    "principal": "Prof Irshad Ali Otho",
+    "rooms": 24,
+    "established": "2003",
+    "contact": {
+        "location": "National Highway N6 bypass Daulatpur",
+        "phone": "+92 300 3003249",
+        "email": "[email protected]",
+        "facebook": "https://www.facebook.com/share/19j9Z1iEvz/"
+    },
+}
 
-def respond(
-    message,
-    history: list[tuple[str, str]],
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-):
-    messages = [{"role": "system", "content": system_message}]
+# Step 3: Custom Question Logic
+def answer_question(question):
+    question = question.lower()
 
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
+    if "college name" in question or "what is the name" in question:
+        return f"The name of the college is {college_data['name']}."
+    
+    elif "about college" in question or "tell me about the college" in question:
+        return f"About the college:\n{college_data['about']}\n\n{college_data['extended_about']}"
+    
+    elif "program" in question or "course" in question:
+        return f"Programs offered:\n- " + "\n- ".join(college_data["programs"])
+    
+    elif "teacher" in question or "faculty" in question or "lecturer" in question:
+        for subject, teachers in college_data["teachers"].items():
+            if subject.lower() in question:
+                return f"The {subject} lecturer(s): " + ", ".join(teachers)
+        return (
+            "Teachers:\n" + "\n".join(
+                [f"{subject}: " + ", ".join(teachers) for subject, teachers in college_data["teachers"].items()]
+            )
+        )
+    
+    elif "room" in question:
+        return f"The college has {college_data['rooms']} rooms available."
+    
+    elif "facility" in question:
+        return f"Facilities available:\n- " + "\n- ".join(college_data["facilities"])
+    
+    elif "principal" in question:
+        return f"Our Principal is {college_data['principal']}."
+    
+    elif "staff" in question or "clerk" in question:
+        clerks = "\n".join(college_data["staff"]["clerks"])
+        lab_assistants = "\n".join(college_data["staff"]["lab_assistants"])
+        other_staff = "\n".join(college_data["staff"]["other"])
+        return f"Clerks:\n{clerks}\n\nLab Assistants:\n{lab_assistants}\n\nOther Staff:\n{other_staff}"
+    
+    elif "location" in question or "where" in question:
+        return f"The college is located at {college_data['contact']['location']}."
+    
+    elif "contact" in question or "email" in question or "phone" in question:
+        contact_info = college_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']}"
+        )
+    
+    elif "establish" in question or "founded" in question or "start" in question:
+        return f"The college was established in {college_data['established']}."
+    
+    else:
+        context = (
+            f"About the college: {college_data['about']}\n"
+            f"Programs offered: {', '.join(college_data['programs'])}\n"
+            f"Teachers: {', '.join([f'{subject}: ' + ', '.join(teachers) for subject, teachers in college_data['teachers'].items()])}\n"
+            f"Facilities: {', '.join(college_data['facilities'])}\n"
+            f"Principal: {college_data['principal']}\n"
+            f"Contact: {college_data['contact']['location']}, {college_data['contact']['phone']}, {college_data['contact']['email']}"
+        )
+        try:
+            result = qa_pipeline(question=question, context=context)
+            if result["score"] > 0.2:
+                return result["answer"]
+        except Exception:
+            pass
 
-    messages.append({"role": "user", "content": message})
+        return (
+            "I'm sorry, I couldn't understand your question. You can ask about programs, teachers, facilities, or contact details."
+        )
 
-    response = ""
+# Step 4: Create Gradio Interface
+with gr.Blocks() as app:
+    gr.HTML("""
+    <style>
+        body {
+            background-color: white;
+            color: black;
+            font-family: Arial, sans-serif;
+        }
+        .gradio-container {
+            background-color: white;
+            border-radius: 10px;
+            padding: 20px;
+            width: 80%;
+            margin: 0 auto;
+        }
+        .gr-textbox input {
+            background-color: #f0f0f0;
+            color: black;
+        }
+        .gr-button {
+            background-color: #007BFF;
+            color: white;
+            border-radius: 5px;
+            padding: 10px;
+            font-size: 16px;
+        }
+        .gr-button:hover {
+            background-color: #0056b3;
+        }
+        .gr-markdown {
+            font-size: 18px;
+        }
+        .gr-textbox {
+            font-size: 22px;
+            height: 150px;
+        }
+        .logo {
+            text-align: center;
+            margin-bottom: 20px;
+        }
+        .partition {
+            border-top: 2px solid #FFD700;
+            margin: 20px 0;
+            padding-top: 10px;
+        }
+        .question-answer-box {
+            transition: background-color 0.3s ease;
+            font-size: 22px;
+            padding: 20px;
+            border-radius: 8px;
+            height: 200px;
+        }
+        .question-answer-box:hover {
+            background-color: #e0f7fa;
+        }
+        .center-text {
+            text-align: center;
+            color: #007BFF;
+        }
+    </style>
+    """)
 
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.choices[0].delta.content
+    # Adding the logo (change the logo path as per your requirement)
+    gr.HTML("""
+    <div class="logo">
+        <img src="https://via.placeholder.com/150" alt="College Logo" />
+    </div>
+    """)
 
-        response += token
-        yield response
+    gr.Markdown("<h1 class='center-text' style='font-size: 36px;'>Government Boys Degree College Daulatpur</h1>")
 
+    # About Section
+    with gr.Row():
+        gr.Markdown("<h3 class='center-text'>About Us</h3>")
+        gr.Markdown(college_data["about"])
 
-"""
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
-"""
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-p (nucleus sampling)",
-        ),
-    ],
-)
+    gr.HTML("<div class='partition'></div>")
+
+    # Contact Information
+    with gr.Row():
+        gr.Markdown("<h3 class='center-text'>Contact Information</h3>")
+        gr.Markdown(f"Location: {college_data['contact']['location']}\nPhone: {college_data['contact']['phone']}\nEmail: {college_data['contact']['email']}")
+
+    gr.HTML("<div class='partition'></div>")
 
+    # Question Answering Section
+    with gr.Row():
+        gr.Markdown("<h3 class='center-text'>Ask Me About the College</h3>")
+        user_input = gr.Textbox(label="Ask your Question:", placeholder="e.g. What programs are offered?", interactive=True)
+        response_output = gr.Textbox(label="Answer", interactive=False, elem_classes="question-answer-box")
+        user_input.submit(answer_question, inputs=user_input, outputs=response_output)
 
-if __name__ == "__main__":
-    demo.launch()
+# Launch the app
+app.launch(share=True)