Update app.py

app.py

@@ -18,25 +18,36 @@ def classify_commit(message):
     # Return the predicted labels as a comma-separated string
     return ", ".join(predicted_labels[0]) if predicted_labels[0] else "No labels"
 
-# Create a Gradio interface with enhanced aesthetics
+# Create a Gradio interface with Dockerfile-specific examples and visuals
 demo = gr.Interface(
     fn=classify_commit,
     inputs=gr.Textbox(
         label="Enter Commit Message",
-        placeholder="Type your commit message here...",
+        placeholder="Type your Dockerfile-related commit message here...",
         lines=3,
         max_lines=5,
     ),
     outputs=gr.Textbox(label="Predicted Labels"),
-    title="🚀 Commit Message Classifier",
-    description="📜 This tool classifies commit messages into categories like 'bug fix', 'feature addition', and more. Enter a message to see the predicted labels.",
+    title="🐳 Dockerfile Commit Message Classifier",
+    description="🔍 Classify commit messages related to Dockerfiles into categories like 'bug fix', 'feature addition', and more. Enter a commit message to get the predictions.",
     examples=[
-        ["Fixed a bug in the login feature"],
-        ["Added a new user dashboard"],
-        ["Updated order processing logic"],
-        ["Refactored the payment module for better performance"],
+        ["Fixed an issue with the base image version in Dockerfile"],
+        ["Added a new multistage build to optimize Docker image size"],
+        ["Updated the Python version in Dockerfile to 3.10"],
+        ["Refactored Dockerfile to reduce build time"],
+        ["Added COPY instruction for configuration files"],
     ],
     theme="compact",  # Optional: A compact Gradio theme
+    article="""
+    **How to Use**:
+    - Enter a commit message related to Dockerfiles.
+    - The tool predicts categories such as 'bug fix', 'code refactoring', 'feature addition', or 'maintenance/other'.
+    
+    **Examples**:
+    - Fixing base image versions.
+    - Adding new Docker build stages.
+    - Updating dependencies in Dockerfiles.
+    """,
 )
 
 # Launch the Gradio app