Update app.py

app.py

@@ -63,7 +63,17 @@ def extract_text_from_link(url):
 
 @st.cache_data(ttl=3600)
 def get_summary(text):
-    prompt = f'''{HUMAN_PROMPT}  
+
+    message = client.messages.create(
+    model = "claude-instant-1.2",
+    max_tokens_to_sample=2000,
+    messages=[
+    {
+        "role": "user",
+        "content": [
+            {
+                "type": "text",
+                "text": 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).
@@ -79,15 +89,9 @@ def get_summary(text):
       - '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.
-    {AI_PROMPT}'''
-    completion = anthropic.completions.create(
-    #model="claude-2",
-    model = "claude-instant-1.2",
-    max_tokens_to_sample=2000,
-    prompt=prompt,
-)
-    response = completion.completion
-    return response
+    '''
+        ]
+    return message.content
     
 search_query = st.text_input("case name, e.g. brown v board supreme, 372 US 335, google v oracle appeal")