Update app.py

app.py

@@ -4,9 +4,8 @@ import requests
 import os
 import anthropic
 from anthropic import Anthropic, HUMAN_PROMPT, AI_PROMPT
-from io import StringIO
-import time 
-from io import BytesIO 
+from io import StringIO, BytesIO
+import time
 
 
 client = anthropic.Anthropic(
@@ -96,8 +95,6 @@ search_query = st.text_input("case name, e.g. brown v board supreme, 372 US 335,
 
 if search_query:
     with st.spinner("Searching for cases..."):
-        st.write(text)
-            
         results = search_legal_cases(search_query)
     if results:
         title, link, citation = results[0]
@@ -106,17 +103,13 @@ if search_query:
         st.write("Citation:\n", citation)
         with st.spinner("Extracting text from case..."):
             text = extract_text_from_link(link)
-        #st.write(text)
+            st.write(text)  # Optionally display the extracted text
+        with st.spinner("Generating summary..."):
+            summary = get_summary(text)
+        st.write(summary)
+        
+        # Convert the response to a file-like object for download
+        download_bytes = BytesIO(summary.encode())
+        st.download_button("Download the results", download_bytes, "summary.txt", "text/plain")
     else:
         st.write("No results found.")
-        
-    
-if search_query:
-    with st.spinner("Generating summary..."):
-        summary = get_summary(text)
-    st.write(summary)
-    
-
-    # Convert the response to a file-like object for download
-    download_bytes = BytesIO(summary.encode())
-    st.download_button("Download the results", download_bytes, "summary.txt", "text/plain")