Update app2.py

app2.py

@@ -581,7 +581,8 @@ def find_case_precedents(case_details: str) -> Dict[str, Any]:
 
         # Compile all information
         compilation_prompt = f"""
-        Analyze and summarize the following case law information, focusing solely on factual elements and legal principles. Do not include any speculative or fictional content:
+        Analyze the following case details and identify key legal concepts and relevant precedents,
+        Analyze and the following case law information, focusing solely on factual elements and legal principles. Do not include any speculative or fictional content:
 
         Case Details: {case_details}
 
@@ -1707,8 +1708,8 @@ elif feature == "Case Precedent Finder":
     if 'precedents' not in st.session_state:
         st.session_state.precedents = None
     
-    case_details = st.text_area("Enter case details:")
-    if st.button("Find Precedents"):
+    case_details = st.text_area("Enter case details:", height=100)
+    if st.button("Find Precedents", type="primary"):
         with st.spinner("Searching for relevant case precedents..."):
             try:
                 st.session_state.precedents = find_case_precedents(case_details)
@@ -1718,32 +1719,51 @@ elif feature == "Case Precedent Finder":
     if st.session_state.precedents:
         precedents = st.session_state.precedents
         
-        st.write("### Summary of Relevant Case Precedents")
-        st.markdown(precedents["summary"])
+        st.markdown("## Summary of Relevant Case Precedents")
+        st.info(precedents["summary"])
         
-        st.write("### Related Cases from Public Databases")
+        st.markdown("## Related Cases from Public Databases")
         for i, case in enumerate(precedents["public_cases"], 1):
-            with st.expander(f"{i}. {case['case_name']}"):
-                st.write(f"**Source:** {case['source']}")
-                st.write(f"**Case Name:** {case['case_name']}")
-                st.write(f"**URL:** [{case['url']}]({case['url']})")
-                
-                # Display all available fields without conditionals
+            st.markdown(f"### {i}. {case['case_name']}")
+            col1, col2 = st.columns([1, 2])
+            with col1:
+                st.markdown(f"**Source:** {case['source']}")
+                st.markdown(f"**URL:** [View Case]({case['url']})")
+            with col2:
                 for field in ['summary', 'date_filed', 'docket_number', 'court']:
                     if field in case and case[field]:
-                        st.write(f"**{field.replace('_', ' ').title()}:** {case[field]}")
+                        st.markdown(f"**{field.replace('_', ' ').title()}:** {case[field]}")
+            st.markdown("---")
         
-        st.write("### Additional Web Results")
+        st.markdown("## Additional Web Results")
         for i, result in enumerate(precedents["web_results"], 1):
-            with st.expander(f"{i}. {result['title']}"):
-                st.write(f"**Source:** [{result['link']}]({result['link']})")
-                st.write(f"**Snippet:** {result['snippet']}")
+            st.markdown(f"### {i}. {result['title']}")
+            st.markdown(f"**Source:** [{result['link']}]({result['link']})")
+            st.markdown(f"**Snippet:** {result['snippet']}")
+            st.markdown("---")
         
-        st.write("### Wikipedia Information")
-        wiki_info = precedents["wikipedia"]
-        with st.expander(f"Wikipedia: {wiki_info['title']}"):
-            st.markdown(wiki_info['summary'])
-            st.write(f"[Read more on Wikipedia]({wiki_info['url']})")
+        st.markdown("## Wikipedia Information")
+        st.markdown(f"### {wiki_info['title']}")
+        st.markdown(wiki_info['summary'])
+        st.markdown(f"[Read more on Wikipedia]({wiki_info['url']})")
+
+    st.markdown(
+        """
+        <style>
+            .stButton > button {
+                width: 100%;
+            }
+            .stTextArea > div > div > textarea {
+                font-size: 16px;
+            }
+            h1, h2, h3 {
+                margin-top: 1em;
+                margin-bottom: 0.5em;
+            }
+        </style>
+        """,
+        unsafe_allow_html=True
+    )
 
 elif feature == "Legal Cost Estimator":
     legal_cost_estimator_ui()