Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from meta_ai_api import MetaAI
+from urllib.parse import urlparse
 
 # Initialize Meta AI API
 ai = MetaAI()
@@ -26,8 +27,12 @@ def fetch_response(query):
 
 def display_sources(sources):
     if sources:
-        for source in sources:
-            st.markdown(f"[{source['title']}]({source['link']})", unsafe_allow_html=True)
+        with st.expander("Show Sources"):
+            for source in sources:
+                # Parse the domain from the URL
+                domain = urlparse(source['link']).netloc
+                # Format and display the domain and title
+                st.markdown(f"- **{domain}**: [{source['title']}]({source['link']})", unsafe_allow_html=True)
     else:
         st.write("No sources available.")
 
@@ -41,11 +46,11 @@ def main():
     image_path = 'img/meta-ai-logo.webp'  # Replace with your image's filename and extension
 
     # Create two columns
-    col1, col2 = st.columns([1, 4])  # Adjust the ratio as needed for your layout
+    col1, col2 = st.columns([1, 2])  # Adjust the ratio as needed for your layout
 
     # Use the first column to display the image
     with col1:
-        st.image(image_path, width=70)
+        st.image(image_path, width=60)
 
     # Use the second column to display the title and other content
     with col2:
@@ -62,7 +67,7 @@ def main():
         st.write(response.get('message', 'No response message.'))
         
         # Display the AI response in a collapsible section
-        with st.expander("Show Full Response"):
+        with st.expander("Show Sources"):
             # Display sources with clickable links in a collapsible section
             display_sources(response.get('sources', []))