Update app.py

app.py

@@ -59,7 +59,7 @@ def fetch_pdfs(city_code):
         st.error(f"Failed to fetch PDFs for city code {city_code}")
         return None
 
-def download_pdf(url):
+def download_pdf(url, doc_title):
     # Add 'https://' scheme if it's missing
     if not url.startswith(('http://', 'https://')):
         url = 'https://' + url
@@ -67,15 +67,21 @@ def download_pdf(url):
     try:
         response = requests.get(url)
         response.raise_for_status()  # Raise an exception for bad status codes
-        
-        # Generate a unique filename
-        filename = f"downloaded_pdf_{st.session_state.id_counter}.pdf"
-        st.session_state.id_counter += 1
-        
+
+        # Sanitize doc_title to create a valid filename
+        sanitized_title = ''.join(c for c in doc_title if c.isalnum() or c in (' ', '_', '-')).rstrip()
+        sanitized_title = sanitized_title.replace(' ', '_')
+        filename = f"{sanitized_title}.pdf"
+
+        # Ensure filename is unique by appending the id_counter if necessary
+        if os.path.exists(filename):
+            filename = f"{sanitized_title}_{st.session_state.id_counter}.pdf"
+            st.session_state.id_counter += 1
+
         # Save the PDF content to a file
         with open(filename, 'wb') as f:
             f.write(response.content)
-        
+
         return filename
     except requests.RequestException as e:
         st.error(f"Failed to download PDF from {url}. Error: {str(e)}")
@@ -206,6 +212,53 @@ def chat_with_assistant(file_ids, user_message):
 
 # ---------------------- Streamlit App ----------------------
 
+# ---------------------- Custom CSS Injection ----------------------
+
+# Inject custom CSS to style chat messages
+st.markdown("""
+    <style>
+    /* Style for the chat container */
+    .chat-container {
+        display: flex;
+        flex-direction: column;
+    }
+
+    /* Style for individual chat messages */
+    .chat-message {
+        margin-bottom: 20px; /* Increased space between messages */
+    }
+
+    /* Style for user messages */
+    .chat-message.user > div:first-child {
+        color: #1E90FF;  /* Dodger Blue for "You" */
+        font-size: 1.2em;
+        margin-bottom: 5px;
+    }
+
+    /* Style for assistant messages */
+    .chat-message.assistant > div:first-child {
+        color: #32CD32;  /* Lime Green for "Assistant" */
+        font-size: 1.2em;
+        margin-bottom: 5px;
+    }
+
+    /* Style for the message content */
+    .message-content {
+        /* Removed the background color to maintain original background */
+        padding: 10px;
+        border-radius: 5px;
+        /* Optionally, you can set a semi-transparent background or match it with your theme */
+        /* background-color: rgba(241, 241, 241, 0.8); */
+    }
+
+    /* Optional: Add more spacing between messages */
+    .chat-message.user, .chat-message.assistant {
+        padding-top: 10px;
+        padding-bottom: 10px;
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
 page = st.sidebar.selectbox("Choose a page", ["Documents", "Home", "Admin"])
 
 if page == "Home":
@@ -223,7 +276,8 @@ if page == "Home":
     # Main chat area improvements
     colored_header("Chat", description="Ask questions about building regulations")
 
-    # Display chat messages with improved styling
+    # Chat container with custom CSS class
+    st.markdown('<div class="chat-container">', unsafe_allow_html=True)
     for chat in st.session_state.chat_history:
         with st.container():
             if chat['role'] == 'user':
@@ -240,6 +294,7 @@ if page == "Home":
                     <div class="message-content">{chat['content']}</div>
                 </div>
                 """, unsafe_allow_html=True)
+    st.markdown('</div>', unsafe_allow_html=True)
 
     # Chat input improvements
     with st.form("chat_form", clear_on_submit=True):
@@ -354,7 +409,8 @@ elif page == "Documents":
                         file_href = pdf['File Href']
                         doc_title = pdf['Doc Title']
                         
-                        file_name = download_pdf(file_href)
+                        # Pass doc_title to download_pdf
+                        file_name = download_pdf(file_href, doc_title)
                         if file_name:
                             file_path = f"./{file_name}"
                             file_id = upload_file_to_openai(file_path)