Update app/ui.py

app/ui.py

@@ -42,7 +42,11 @@ def get_base64_image(image_path):
 st.set_page_config(
     page_title="DTECT",
     page_icon="🔍",
-    layout="wide"
+    layout="wide",
+    menu_items={
+        'Get Help': 'https://github.com/AdhyaSuman/DTECT',
+        'Report a bug': "https://github.com/AdhyaSuman/DTECT/issues/new"
+    }
 )
 
 # Sidebar branding and repo link
@@ -65,8 +69,7 @@ AVAILABLE_MODELS = ["DTM", "DETM", "CFDTM"]
 ENV_VAR_MAP = {
     "OpenAI": "OPENAI_API_KEY",
     "Anthropic": "ANTHROPIC_API_KEY",
-    "Gemini": "GEMINI_API_KEY",
-    "Mistral": "MISTRAL_API_KEY"
+    "Gemini": "GEMINI_API_KEY"
 }
 
 def list_datasets(data_dir):
@@ -261,8 +264,10 @@ if selected_words:
             legendgroup=word,
             showlegend=True
         ))
-    fig.update_layout(title="", xaxis_title="Year", yaxis_title="Importance")
-    st.plotly_chart(fig, use_container_width=True)
+    fig.update_layout(title="", xaxis_title="Year", yaxis_title="Importance", legend=dict(font=dict(size=16)))
+    _, chart_col, _ = st.columns([0.2, 0.6, 0.2])
+    with chart_col:
+        st.plotly_chart(fig, use_container_width=True)
 
 # ==============================================================================
 # 3. 🔍 DOCUMENT RETRIEVAL & 📃 SUMMARIZATION