Update app.py

app.py

@@ -19,18 +19,7 @@ st.set_page_config(
 
 
 def main():
-    # Sidebar navigation
-    st.sidebar.title("SuoMoto.AI ⚖️")
-    st.sidebar.markdown(
-        """
-        Streamline legal workflows with advanced AI-powered tools for document management, chat, and template generation.
-        """
-    )
-    tab = st.sidebar.radio(
-        "Navigation",
-        ["📁 Case Manager", "📄 Document Manager", "🤖 Chat", "📝 Generate Templates"]
-    )
-
+    # Header (placed at the start of the app)
     st.markdown(
         """
         <style>
@@ -60,6 +49,18 @@ def main():
         unsafe_allow_html=True
     )
 
+    # Sidebar navigation
+    st.sidebar.title("SuoMoto.AI ⚖️")
+    st.sidebar.markdown(
+        """
+        Streamline legal workflows with advanced AI-powered tools for document management, chat, and template generation.
+        """
+    )
+    tab = st.sidebar.radio(
+        "Navigation",
+        ["📁 Case Manager", "📄 Document Manager", "🤖 Chat", "📝 Generate Templates"]
+    )
+
     # Tab 1: Case Manager
     if tab == "📁 Case Manager":
         st.title("📁 Manage Cases")
@@ -154,4 +155,33 @@ def main():
                 with st.spinner("Generating response..."):
                     context = selected_document['text']
                     response = vector_store.chat_with_context(query, context)
-                    st.markdow
+                    st.markdown(f"**Response**: {response}")
+
+                # Allow adding new documents mid-chat
+                uploaded_file = st.file_uploader("📂 Upload Document During Chat", key="chat_upload")
+                if uploaded_file:
+                    with st.spinner("Processing document..."):
+                        text, chunks, metadata = doc_processor.process_and_tag_document(uploaded_file)
+                        doc_data = {"title": uploaded_file.name, "text": text, "metadata": metadata}
+                        case_manager.add_document(selected_case, doc_data)
+                        st.success(f"Document '{uploaded_file.name}' added to case!")
+
+    # Tab 4: Generate Templates
+    elif tab == "📝 Generate Templates":
+        st.title("📝 Legal Document Templates")
+        st.markdown("Generate standard legal documents using predefined templates.")
+        render_template_generator()
+
+    # Footer
+    st.markdown(
+        """
+        <div class="footer">
+        Synaptyx: SuoMoto.AI agent
+        </div>
+        """,
+        unsafe_allow_html=True
+    )
+
+
+if __name__ == "__main__":
+    main()