Update app.py

app.py

@@ -13,10 +13,42 @@ from transformers import AutoModel, AutoTokenizer, pipeline
 import joblib
 from typing import List
 
+# Define styles for different sections
+custom_styles = """
+<style>
+    [id^=tabs-] {
+        padding: 20px;
+        border: 1px solid #ccc;
+        margin-bottom: 20px;
+        box-shadow: 0 4px 8px rgba(0,0,0,0.1);
+    }
+
+    .tab h2 {
+        font-size: 24px;
+        margin-top: 0;
+        color: black;
+    }
+
+    .response-container {
+        padding: 20px;
+        border: 1px solid #ccc;
+        background-color: #f9f9f9;
+        margin-bottom: 20px;
+        box-shadow: 0 4px 8px rgba(0,0,0,0.1);
+    }
+
+    .response-container h3 {
+        font-size: 20px;
+        margin-top: 0;
+        color: black;
+    }
+</style>
+"""
+
 def setup_page():
     st.set_page_config(page_title="EAD Generation", layout="wide")
     display_app_header()
-    
+    st.markdown(custom_styles, unsafe_allow_html=True)
     # Create two columns - one for query history and one for main content
     col1, col2 = st.columns([1, 4])
     
@@ -33,14 +65,14 @@ def display_app_header():
     """Display the header of the Streamlit app."""
     st.title("EAD Generation")
     #st.subheader("Tries")
-    st.markdown("---")
+        st.markdown("---")
     # Add a description of the app
-    st.markdown("""This app allows you to generate EAD/XML archival descriptions. See this serie of blog posts for explanations : 
-                
+    st.markdown("""This app allows you to generate EAD/XML archival descriptions. See this serie of blog posts for explanations :""")                
+    st.markdown("""
     - [https://iaetbibliotheques.fr/2024/11/comment-apprendre-lead-a-un-llm](https://iaetbibliotheques.fr/2024/11/comment-apprendre-lead-a-un-llm)
     - [https://iaetbibliotheques.fr/2024/11/comment-apprendre-lead-a-un-llm-rag-23](https://iaetbibliotheques.fr/2024/11/comment-apprendre-lead-a-un-llm-rag-23)
     - [https://iaetbibliotheques.fr/2024/12/comment-apprendre-lead-a-un-llm-fine-tuning-33](https://iaetbibliotheques.fr/2024/12/comment-apprendre-lead-a-un-llm-fine-tuning-33)
-    """)
+    """, unsafe_allow_html=True)
     st.markdown("---")
 
 # Display the header of the app and get columns