Update app.py

app.py

@@ -10,10 +10,64 @@ except ImportError as e:
     st.error("Please make sure all required libraries are installed.")
     st.stop()
 
+def set_custom_style():
+    st.markdown("""
+    <style>
+    .stApp {
+        max-width: 800px;
+        margin: 0 auto;
+        background-color: #f0f2f6;
+        padding: 2rem;
+        border-radius: 10px;
+    }
+    .title {
+        color: #1e3a8a;
+        font-size: 2.5rem;
+        font-weight: bold;
+        margin-bottom: 2rem;
+        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;
+        font-weight: bold;
+        padding: 0.5rem 2rem;
+        border-radius: 5px;
+        border: none;
+        transition: background-color 0.3s ease;
+    }
+    .stButton>button:hover {
+        background-color: #2563eb;
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
 def main():
-    st.title("Image Captioning App")
+    set_custom_style()
+    
+    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:
@@ -24,8 +78,10 @@ def main():
                 with st.spinner('Generating caption...'):
                     try:
                         caption = generate_caption(image)
-                        st.write("Generated Caption:")
-                        st.write(caption)
+                        st.markdown('<div class="caption-section">', 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)
                     except Exception as e:
                         st.error(f"An error occurred while generating the caption: {str(e)}")
                         st.error("Detailed error traceback:")