Update app.py

app.py

@@ -1,13 +1,6 @@
 import gradio as gr
-from bs4 import BeautifulSoup
 import requests
-import os
-
-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"
-}
-
-proxies = {"http": os.getenv("HTTP_PROXY")}
+from bs4 import BeautifulSoup
 
 def search_legal_cases(query, num_results=10):
     url = "https://scholar.google.com/scholar?hl=en&as_sdt=6"
@@ -22,7 +15,7 @@ def search_legal_cases(query, num_results=10):
         "as_sdt": "4",  # This parameter filters the search results to legal cases
     }
 
-    response = requests.get(url, proxies=proxies, headers=headers, params=params)
+    response = requests.get(url, headers=headers, params=params)
     soup = BeautifulSoup(response.text, "html.parser")
 
     results = []
@@ -35,23 +28,15 @@ def search_legal_cases(query, num_results=10):
 
     return results
 
-def main(search_query):
-    results = search_legal_cases(search_query)
+def get_legal_cases(query):
+    results = search_legal_cases(query)
     if results:
-        title, link, citation = results[0]
-        return title, citation, link
+        response = ""
+        for title, link, citation in results:
+            response += f"**Title:** {title}\n**Link:** {link}\n**Citation:** {citation}\n\n"
+        return response
     else:
         return "No results found."
 
-def display_results(search_query):
-    title, citation, link = main(search_query)
-    return f"Title:\n{title}\n\nCitation:\n{citation}\n\nLink:\n{link}"
-
-iface = gr.Interface(
-    fn=display_results,
-    inputs=gr.inputs.Textbox(lines=1, placeholder="Enter case name, e.g. brown v board supreme, 372 US 335, google v oracle appeal"),
-    outputs=gr.outputs.Textbox(label="Results"),
-    live=True
-)
-
-iface.launch()
+demo = gr.Interface(fn=get_legal_cases, inputs="text", outputs="markdown")
+demo.launch()