Upload app.py

app.py

@@ -31,36 +31,38 @@ def calculate_ranking(data):
         institution["Rank"] = rank
     return ranked_data
 
-# Chatbot function with ranking logic
+# Predefined ranking data
+example_data = [
+    {"Institution": "A", "TLR": 70, "GO": 85, "OI": 90, "PR": 75},
+    {"Institution": "B", "TLR": 80, "GO": 88, "OI": 85, "PR": 90},
+    {"Institution": "C", "TLR": 65, "GO": 80, "OI": 70, "PR": 60},
+]
+
+# Chatbot function
 def chatbot_response(user_message):
+    # Check for predefined data queries
     if "rank" in user_message.lower():
-        # Example data for ranking
-        example_data = [
-            {"Institution": "A", "TLR": 70, "GO": 85, "OI": 90, "PR": 75},
-            {"Institution": "B", "TLR": 80, "GO": 88, "OI": 85, "PR": 90},
-            {"Institution": "C", "TLR": 65, "GO": 80, "OI": 70, "PR": 60},
-        ]
         ranked_data = calculate_ranking(example_data)
         response = "Here are the ranks of the institutions:\n"
         for institution in ranked_data:
             response += f"Rank {institution['Rank']}: {institution['Institution']} (Total Score: {institution['Total']})\n"
         return response
-    else:
-        # Generate chatbot response from model
-        outputs = pipeline(
-            user_message,
-            max_new_tokens=256,
-            do_sample=True,
-            temperature=0.6,
-            top_p=0.9,
-        )
-        return outputs[0]["generated_text"]
+    
+    # Fallback to model-generated response for out-of-scope questions
+    outputs = pipeline(
+        user_message,
+        max_new_tokens=100,  # Restrict length for unexpected questions
+        do_sample=True,
+        temperature=0.7,  # Slightly random responses for more natural output
+        top_p=0.9,
+    )
+    return outputs[0]["generated_text"]
 
 # Gradio interface
 def build_gradio_ui():
     with gr.Blocks() as demo:
-        gr.Markdown("## Chatbot with Hugging Face Spaces")
-        gr.Markdown("Type a message to interact with the chatbot! (Ask about institution rankings too!)")
+        gr.Markdown("## Chatbot with Predefined Data and AI Responses")
+        gr.Markdown("Ask about institution rankings or any other general query!")
         with gr.Row():
             user_input = gr.Textbox(label="Your Message", placeholder="Type your message here...")
             chatbot_output = gr.Textbox(label="Chatbot Response", interactive=False)