Update app2.py

app2.py

@@ -451,25 +451,41 @@ def query_public_case_law(query: str) -> List[Dict[str, Any]]:
         justia_response = requests.get(justia_url, headers=justia_headers)
         justia_response.raise_for_status()
         justia_soup = BeautifulSoup(justia_response.text, 'html.parser')
+        
         justia_results = justia_soup.find_all('div', class_='g')
         
-        for result in justia_results[:3]:  # Limit to 3 results
-            title_elem = result.find('h3', class_='r')
-            if title_elem and 'law.justia.com' in title_elem.find('a')['href']:
-                title = title_elem.text.strip()
-                url = title_elem.find('a')['href']
-                snippet = result.find('div', class_='s').text.strip()
+        for result in justia_results[:5]:  # Limit to top 5 results
+            title_elem = result.find('h3')
+            link_elem = result.find('a')
+            snippet_elem = result.find('div', class_='VwiC3b')
+            
+            if title_elem and link_elem and snippet_elem:
+                title = title_elem.text
+                link = link_elem['href']
+                snippet = snippet_elem.text
+                
+                # Extract case name and citation from the title
+                case_info = title.split(' - ')
+                if len(case_info) >= 2:
+                    case_name = case_info[0]
+                    citation = case_info[1]
+                else:
+                    case_name = title
+                    citation = "Citation not found"
+                
                 cases.append({
                     "source": "Justia",
-                    "case_name": title,
+                    "case_name": case_name,
+                    "citation": citation,
                     "summary": snippet,
-                    "url": url
+                    "url": link
                 })
     except requests.RequestException as e:
         print(f"Error querying Justia: {e}")
 
     # CourtListener Search
     courtlistener_url = f"https://www.courtlistener.com/api/rest/v3/search/?q={query}&type=o&format=json"
+    courtlistener_data = {}
     for attempt in range(3):  # Retry up to 3 times
         try:
             courtlistener_response = requests.get(courtlistener_url)
@@ -480,7 +496,6 @@ def query_public_case_law(query: str) -> List[Dict[str, Any]]:
             print(f"Attempt {attempt + 1} failed: {e}")
             if attempt == 2:
                 print(f"Failed to retrieve or parse data from CourtListener: {e}")
-                break
             time.sleep(2)
 
     if 'results' in courtlistener_data: