Update app.py

app.py

@@ -174,9 +174,39 @@ def generate_prediction(scenario_text, max_length=300, temperature=0.7):
 
 def create_interface():
     """Create the Gradio interface"""
+     # Custom CSS for better styling
+    css = """
+    .gradio-container {
+        font-family: 'Arial', sans-serif;
+    }
+    .header {
+        text-align: center;
+        margin-bottom: 30px;
+    }
+    .warning-box {
+        background-color: #fff3cd;
+        border: 1px solid #ffeaa7;
+        border-radius: 5px;
+        padding: 15px;
+        margin: 10px 0;
+    }
+    .error-box {
+        background-color: #f8d7da;
+        border: 1px solid #f5c6cb;
+        border-radius: 5px;
+        padding: 15px;
+        margin: 10px 0;
+        color: #721c24;
+    }
+    .success-box {
+        background-color: #d4edda;
+        border: 1px solid #c3e6cb;
+        border-radius: 5px;
         padding: 15px;
         margin: 10px 0;
+        color: #155724;
     }
+    """
     
     with gr.Blocks(css=css, title="Workplace Safety Risk Predictor") as interface:
                         temperature = gr.Slider(