Update app.py

app.py

@@ -146,51 +146,18 @@ def chat_about_image(message, chat_history):
         return "", chat_history + [(message, f"Error: {str(e)}")]
 
 
+# Custom CSS for improved styling
 custom_css = """
-.container {
-    max-width: 1000px;
-    margin: auto;
-    padding-top: 1.5rem;
-}
-.header {
-    text-align: center;
-    margin-bottom: 2rem;
-}
-.header h1 {
-    color: #2c3e50;
-    font-size: 2.5rem;
-}
-.subheader {
-    color: #34495e;
-    font-size: 1.2rem;
-    margin-bottom: 2rem;
-}
-.image-container {
-    border: 2px dashed #3498db;
-    border-radius: 10px;
-    padding: 1rem;
-    text-align: center;
-}
-.analyze-button {
-    background-color: #2ecc71 !important;
-    color: white !important;
-}
-.clear-button {
-    background-color: #e74c3c !important;
-    color: white !important;
-}
-.chatbot {
-    border: 1px solid #bdc3c7;
-    border-radius: 10px;
-    padding: 1rem;
-    height: 400px;
-    overflow-y: auto;
-}
-.chat-input {
-    border: 1px solid #bdc3c7;
-    border-radius: 5px;
-    padding: 0.5rem;
-}
+.container { max-width: 1000px; margin: auto; padding-top: 1.5rem; }
+.header { text-align: center; margin-bottom: 2rem; }
+.header h1 { color: #2c3e50; font-size: 2.5rem; }
+.subheader { color: #34495e; font-size: 1.2rem; margin-bottom: 2rem; }
+.image-container { border: 2px dashed #3498db; border-radius: 10px; padding: 1rem; text-align: center; }
+.analyze-button { background-color: #2ecc71 !important; color: white !important; }
+.clear-button { background-color: #e74c3c !important; color: white !important; }
+.chatbot { border: 1px solid #bdc3c7; border-radius: 10px; padding: 1rem; height: 400px; overflow-y: auto; }
+.chat-input { border: 1px solid #bdc3c7; border-radius: 5px; padding: 0.5rem; }
+.groq-badge { position: fixed; bottom: 10px; right: 10px; background-color: #f39c12; color: white; padding: 5px 10px; border-radius: 5px; font-weight: bold; }
 """
 
 # Create the Gradio interface
@@ -199,23 +166,23 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as iface:
         """
         <div class="container">
             <div class="header">
-                <h1>🏗️ Construction Image Analyzer with AI Chat</h1>
+                <h1>🏗️ Construction Site Safety Analyzer</h1>
             </div>
-            <p class="subheader">Upload a construction site image, analyze it for issues, and chat with AI about the findings.</p>
+            <p class="subheader">Enhance workplace safety and compliance with AI-powered image analysis and expert chat assistance.</p>
         </div>
         """
     )
     
     with gr.Row():
         with gr.Column(scale=1):
-            image_input = gr.Image(type="pil", label="Upload Construction Image", elem_classes="image-container")
-            analyze_button = gr.Button("🔍 Analyze Image", elem_classes="analyze-button")
+            image_input = gr.Image(type="pil", label="Upload Construction Site Image", elem_classes="image-container")
+            analyze_button = gr.Button("🔍 Analyze Safety Hazards", elem_classes="analyze-button")
         with gr.Column(scale=2):
-            chatbot = gr.Chatbot(label="Analysis Results and Chat", elem_classes="chatbot")
+            chatbot = gr.Chatbot(label="Safety Analysis Results and Expert Chat", elem_classes="chatbot")
             with gr.Row():
                 msg = gr.Textbox(
-                    label="Ask a question about the image",
-                    placeholder="Type your question here and press Enter...",
+                    label="Ask about safety measures or regulations",
+                    placeholder="E.g., 'What OSHA guidelines apply to this hazard?'",
                     show_label=False,
                     elem_classes="chat-input"
                 )
@@ -230,6 +197,12 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as iface:
     msg.submit(chat_about_image, [msg, chatbot], [msg, chatbot])
     clear.click(lambda: None, None, chatbot, queue=False)
 
+    gr.HTML(
+        """
+        <div class="groq-badge">Powered by Groq</div>
+        """
+    )
+
 # Launch the app
 if __name__ == "__main__":
     iface.launch(debug=True)