app.py

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")}

def search_legal_cases(query, num_results=10):
    url = "https://scholar.google.com/scholar?hl=en&as_sdt=6"
    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"
    }

    params = {
        "q": query,
        "hl": "en",
        "num": num_results,
        "as_sdt": "4",  # This parameter filters the search results to legal cases
    }

    response = requests.get(url, proxies=proxies, headers=headers, params=params)
    soup = BeautifulSoup(response.text, "html.parser")

    results = []
    for result in soup.find_all("div", class_="gs_ri"):
        title = result.find("h3", class_="gs_rt").text
        base_url = "https://scholar.google.com"
        link = base_url + result.find("a")["href"]
        citation = result.find("div", class_="gs_a").text.replace(" - Google Scholar", "")
        results.append((title, link, citation))

    return results

def main(search_query):
    results = search_legal_cases(search_query)
    if results:
        title, link, citation = results[0]
        return title, citation, link
    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()