Update app.py

app.py

@@ -2,15 +2,8 @@ import streamlit as st
 from bs4 import BeautifulSoup
 import requests
 import os
-import anthropic
-from anthropic import Anthropic, HUMAN_PROMPT, AI_PROMPT
-from io import StringIO, BytesIO
 import time
-
-
-client = anthropic.Anthropic(
-   api_key="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", 
-)
+from openai import OpenAI
 
 
 headers = {
@@ -20,6 +13,7 @@ headers = {
 proxies = {"http": os.getenv("HTTP_PROXY")}
 
 
+
 @st.cache_data(ttl=3600)
 def search_legal_cases(query, num_results=10):
     url = "https://scholar.google.com/scholar?hl=en&as_sdt=6"
@@ -62,33 +56,28 @@ def extract_text_from_link(url):
 
 @st.cache_data(ttl=3600)
 def get_summary(text):
-    message = client.messages.create(
-        model="claude-3-opus-20240229",
-        max_tokens_to_sample=2000,
-        messages=[
-            {
-                "role": "user",
-                "content": f'''  
-                You are a law professor specialized in legal writing and legal research. 
-                Summarize the case in {text} in json format according to the following requirements:  
-                Facts (name of the case and its parties, what happened factually).
-                Procedural history (what happened in the past procedurally, what were prior judgements). 
-                Issues (what is in dispute).
-                Holding (the applied rule of law). 
-                Rationale (reasons for the holding).
-                Decision (what did the court decide, e.g. affirmed, overruled).
-                Other opinions (if there are any dissenting or concurring opinions, summarize majority opinion, dissenting opinion and concurring opinion).
-                Cases cited (which cases the court cited and how it treated them): 
-                  - 'red flag' means some negative treatment — such as a cited case is being overruled, superseded, or not followed by this court for some reason.
-                  - 'yellow flag' means the case has some negative treatment by this court but has not been reversed or overruled. For example, the reasoning of the decision was criticized or its holding was limited to a specific set of facts.
-                  - 'blue flag' means the case has been appealed to the U.S. Court of Appeals or the U.S. Supreme Court (excluding appeals originating from agencies).
-
-                Present the summary in json format.
-                '''
-            }
-        ]
-    )
-    return message.content
+   client = OpenAI(api_key='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')
+
+   completion = client.chat.completions.create(
+     model="gpt-4o",
+     messages=[
+       {"role": "system", "content": f'''You are a law professor specialized in legal writing and legal research. 
+                   When presented with a case by a user please summarize it according to the following requirements:
+                   Name of the court.
+                   Facts (name of the parties, what happened factually).
+                   Procedural history (what happened in the past procedurally, what were prior judgements). 
+                   Issues (what is in dispute).
+                   Holding (the applied rule of law). 
+                   Rationale (reasons for the holding).
+                   Decision (what did the court decide, e.g. affirmed, overruled).
+                   Other opinions (if there are any dissenting or concurring opinions, summarize majority opinion, dissenting opinion and concurring opinion).
+                   Cases cited (which cases the court cited and how it treated them)'''},
+       {"role": "user", "content": f"Summarize this case: {text}!"}
+     ]
+   )
+   
+   return completion.choices[0].message.content
+
 
 st.write("\n")
 st.write("\n")
@@ -102,15 +91,12 @@ if search_query:
         st.write("Title:\n", title)
         #st.write("Link:\n", link)
         st.write("Citation:\n", citation)
-        with st.spinner("Extracting text from case..."):
-            text = extract_text_from_link(link)
-            st.write(text)  # Optionally display the extracted text
-        with st.spinner("Generating summary..."):
-            summary = get_summary(text)
+        #with st.spinner("Extracting text from case / Generating summary"):
+        text = extract_text_from_link(link)
+        #st.write(text)  # Optionally display the extracted text
+        
+        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.")
+        st.write("No results found.")