Update app.py

app.py

@@ -16,35 +16,15 @@ def set_custom_style():
     .stApp {
         max-width: 800px;
         margin: 0 auto;
-        background-color: #f0f2f6;
-        padding: 2rem;
-        border-radius: 10px;
+        padding: 1rem;
     }
     .title {
         color: #1e3a8a;
         font-size: 2.5rem;
         font-weight: bold;
-        margin-bottom: 2rem;
+        margin-bottom: 1rem;
         text-align: center;
     }
-    .upload-section {
-        background-color: white;
-        padding: 2rem;
-        border-radius: 10px;
-        box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-    }
-    .caption-section {
-        background-color: white;
-        padding: 2rem;
-        border-radius: 10px;
-        box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-        margin-top: 2rem;
-    }
-    .caption-text {
-        font-size: 1.2rem;
-        font-style: italic;
-        color: #4a5568;
-    }
     .stButton>button {
         background-color: #3b82f6;
         color: white;
@@ -57,6 +37,24 @@ def set_custom_style():
     .stButton>button:hover {
         background-color: #2563eb;
     }
+    .caption-container {
+        background-color: #f3f4f6;
+        padding: 1rem;
+        border-radius: 5px;
+        margin-top: 1rem;
+    }
+    .caption-text {
+        font-size: 1.2rem;
+        font-style: italic;
+        color: #4a5568;
+    }
+    .stAlert {
+        background-color: #fee2e2;
+        color: #991b1b;
+        padding: 0.5rem;
+        border-radius: 5px;
+        margin-top: 1rem;
+    }
     </style>
     """, unsafe_allow_html=True)
 
@@ -65,9 +63,7 @@ def main():
     
     st.markdown('<h1 class="title">Image Captioning App</h1>', unsafe_allow_html=True)
 
-    st.markdown('<div class="upload-section">', unsafe_allow_html=True)
     uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
-    st.markdown('</div>', unsafe_allow_html=True)
     
     if uploaded_file is not None:
         try:
@@ -78,7 +74,7 @@ def main():
                 with st.spinner('Generating caption...'):
                     try:
                         caption = generate_caption(image)
-                        st.markdown('<div class="caption-section">', unsafe_allow_html=True)
+                        st.markdown('<div class="caption-container">', unsafe_allow_html=True)
                         st.markdown("### Generated Caption:")
                         st.markdown(f'<p class="caption-text">{caption}</p>', unsafe_allow_html=True)
                         st.markdown('</div>', unsafe_allow_html=True)