Upload app.py

app.py

@@ -1,11 +1,13 @@
 import gradio as gr
 import requests
+import json  # Import json module to parse response text
 
 # Function to call the "Previous Cases" API (cts endpoint)
 def previous_cases(query):
     try:
         response = requests.post("https://aaronanoronha-haha.hf.space/cts", json={"topic": query})
-        return response.text
+        data = json.loads(response.text)  # Parse JSON string into dictionary
+        return data.get("response", "No response found")  # Extract 'response' field
     except Exception as e:
         return f"Error: {e}"
 
@@ -13,7 +15,8 @@ def previous_cases(query):
 def law_navigation(query):
     try:
         response = requests.post("https://aaronanoronha-haha.hf.space/lsi", json={"topic": query})
-        return response.text
+        data = json.loads(response.text)
+        return data.get("response", "No response found")
     except Exception as e:
         return f"Error: {e}"
 
@@ -21,7 +24,8 @@ def law_navigation(query):
 def judgment_prediction(query):
     try:
         response = requests.post("https://aaronanoronha-haha.hf.space/pcr", json={"topic": query})
-        return response.text
+        data = json.loads(response.text)
+        return data.get("response", "No response found")
     except Exception as e:
         return f"Error: {e}"
 
@@ -29,7 +33,8 @@ def judgment_prediction(query):
 def bail_analysis(query):
     try:
         response = requests.post("https://aaronanoronha-haha.hf.space/bail", json={"topic": query})
-        return response.text
+        data = json.loads(response.text)
+        return data.get("response", "No response found")
     except Exception as e:
         return f"Error: {e}"