Update app.py

app.py

@@ -597,4 +597,51 @@ def get_url_params(request: gr.Request):
     return "Load Distributor", None
 
 def handle_user_access(request: gr.Request):
-    """Handle user access an
+    """Handle user access and perform load balancing"""
+    title, check_id = get_url_params(request)
+    
+    if not check_id:
+        # No student ID provided
+        error_html = """
+        <div style="max-width: 600px; margin: 50px auto; padding: 30px; text-align: center; 
+                    background: #fff3cd; border: 1px solid #ffeaa7; border-radius: 12px;">
+            <h2 style="color: #856404;">⚠️ Invalid Access</h2>
+            <p style="color: #856404; font-size: 16px; line-height: 1.6;">
+                This load distributor requires a valid student ID parameter.<br><br>
+                <strong>Please access this system through the official link provided in Moodle.</strong>
+            </p>
+        </div>
+        """
+        return title, gr.HTML(error_html)
+    
+    # Valid student ID - perform load balancing
+    try:
+        result = load_balance_user(check_id)
+        return title, result
+    except Exception as e:
+        # Handle any errors during load balancing
+        error_html = f"""
+        <div style="max-width: 600px; margin: 50px auto; padding: 30px; text-align: center; 
+                    background: #f8d7da; border: 1px solid #f5c6cb; border-radius: 12px;">
+            <h2 style="color: #721c24;">🚫 Load Balancing Error</h2>
+            <p style="color: #721c24; font-size: 16px; line-height: 1.6;">
+                {str(e)}<br><br>
+                Please try again in a few moments or contact your instructor if the problem persists.
+            </p>
+        </div>
+        """
+        return title, gr.HTML(error_html)
+
+# Create Gradio interface
+with gr.Blocks(title="CIV3283 Load Distributor") as interface:
+    title_display = gr.Markdown("# 🔄 CIV3283 Learning Assistant Load Distributor", elem_id="title")
+    content_display = gr.HTML("")
+    
+    # Initialize on page load
+    interface.load(
+        fn=handle_user_access,
+        outputs=[title_display, content_display]
+    )
+
+if __name__ == "__main__":
+    interface.launch()