Update app.py

app.py

@@ -4,7 +4,7 @@ import os
 from openai import OpenAI
 
 headers = {
-    "User-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.102 Safari/537.36 Edge/18.19582"
+    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.102 Safari/537.36 Edge/18.19582"
 }
 
 proxies = {"http": os.getenv("HTTP_PROXY")}
@@ -13,17 +13,18 @@ proxies = {"http": os.getenv("HTTP_PROXY")}
 def extract_text_from_api(query):
     url = "https://fastapi-example-uqyy.onrender.com/extract_text"
     headers = {
-        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/93.0.4577.63 Safari/537.3"
+        "accept": "application/json"
     }
     
     params = {
-        "query": query
+        "query": query,
+        "num_results": 1
     }
     
     response = requests.get(url, headers=headers, proxies=proxies, params=params)
     
     if response.status_code == 200:
-        return response.json().get("extracted_text", "")
+        return response.json().get("text", "")
     else:
         return ""
 
@@ -32,7 +33,7 @@ def get_summary(text):
     client = OpenAI(api_key='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')
 
     completion = client.chat.completions.create(
-        model="gpt-4o",
+        model="gpt-4",
         messages=[
             {"role": "system", "content": '''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:
@@ -54,7 +55,7 @@ def get_summary(text):
 
 st.write("\n")
 st.write("\n")
-search_query = st.text_input("Enter the case query, e.g., 'brown v board supreme'")
+search_query = st.text_input("Enter the case query, e.g., 'Google oracle supreme'")
 
 if search_query:
     with st.spinner("Extracting text from case..."):