ositamiles commited on
Commit
2ea9d08
·
verified ·
1 Parent(s): 50e94c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -6
app.py CHANGED
@@ -4,12 +4,22 @@ import requests
4
  # API URL
5
  API_URL = "https://startrz-devi.hf.space/api/v1/prediction/e54adffc-ae77-42e5-9fc0-c4584e081093"
6
 
7
- # Function to query the API
8
  def query(payload):
9
  try:
10
  response = requests.post(API_URL, json=payload)
11
  response.raise_for_status() # Raise an error for bad HTTP responses
12
- return response.json() # Return the JSON response
 
 
 
 
 
 
 
 
 
 
13
  except requests.exceptions.RequestException as e:
14
  return {"error": str(e)}
15
 
@@ -30,11 +40,19 @@ def main():
30
  "question": user_input,
31
  })
32
 
33
- # Display the raw JSON response
34
- st.subheader("Raw JSON Response:")
35
- st.json(response)
 
 
 
 
 
 
 
 
36
  else:
37
  st.warning("Please enter a question before submitting.")
38
 
39
  if __name__ == "__main__":
40
- main()
 
4
  # API URL
5
  API_URL = "https://startrz-devi.hf.space/api/v1/prediction/e54adffc-ae77-42e5-9fc0-c4584e081093"
6
 
7
+ # Function to query the API and extract toolOutput
8
  def query(payload):
9
  try:
10
  response = requests.post(API_URL, json=payload)
11
  response.raise_for_status() # Raise an error for bad HTTP responses
12
+ data = response.json() # Parse the JSON response
13
+
14
+ # Extract toolOutput if present in the response
15
+ tool_outputs = []
16
+ if "agentReasoning" in data:
17
+ for reasoning in data["agentReasoning"]:
18
+ if "usedTools" in reasoning:
19
+ tool_outputs.extend(reasoning["usedTools"])
20
+
21
+ # Return the full response and toolOutput
22
+ return {"raw_response": data, "tool_output": tool_outputs}
23
  except requests.exceptions.RequestException as e:
24
  return {"error": str(e)}
25
 
 
40
  "question": user_input,
41
  })
42
 
43
+ # Display the extracted toolOutput on the main page
44
+ st.subheader("Extracted Tool Output:")
45
+ tool_output = response.get("tool_output", [])
46
+ if tool_output:
47
+ st.write(tool_output)
48
+ else:
49
+ st.write("No toolOutput data found in the response.")
50
+
51
+ # Display the raw JSON response in the sidebar
52
+ st.sidebar.header("Raw JSON Response")
53
+ st.sidebar.json(response.get("raw_response", {}))
54
  else:
55
  st.warning("Please enter a question before submitting.")
56
 
57
  if __name__ == "__main__":
58
+ main()