umm-maybe commited on
Commit
de2ebc5
·
1 Parent(s): 1baddb8
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -16,7 +16,8 @@ def vizTour(activityList):
16
  net.add_edge(lastActivity, activity)
17
  lastActivity = activity
18
  net.save_graph('tour.html')
19
- return True
 
20
 
21
  def getTour(income,size,years,sex,edu,wrk):
22
  # person = a dict with person-level and hh-level attributes:
@@ -82,8 +83,8 @@ def getTour(income,size,years,sex,edu,wrk):
82
  print(e)
83
  else:
84
  print("Nothing extracted!")
85
- success = vizTour(activity_list)
86
- return activity_list, success
87
 
88
  with gr.Interface(fn=getTour, inputs=[
89
  gr.inputs.Textbox(label="Annual Household Income (in dollars)"),
@@ -92,6 +93,5 @@ with gr.Interface(fn=getTour, inputs=[
92
  gr.inputs.Dropdown(["unknown", "male", "female"], label="Gender/sex"),
93
  gr.inputs.Dropdown(["unknown", "grade school","highschool", "associates", "bachelors", "graduate"], label="Educational attainment level"),
94
  gr.inputs.Dropdown(["unknown", "yes","no"], label="Worker status")],
95
- outputs=["json",
96
- gr.HTML(value=open('tour.html'))], title="GPT-Tour", description="Generate a tour for a person", allow_flagging=False, allow_screenshot=False, allow_embedding=False) as iface:
97
  iface.launch()
 
16
  net.add_edge(lastActivity, activity)
17
  lastActivity = activity
18
  net.save_graph('tour.html')
19
+ tour_html = open('tour.html', 'r')
20
+ return tour_html
21
 
22
  def getTour(income,size,years,sex,edu,wrk):
23
  # person = a dict with person-level and hh-level attributes:
 
83
  print(e)
84
  else:
85
  print("Nothing extracted!")
86
+ tour_html = vizTour(activity_list)
87
+ return activity_list, tour_html
88
 
89
  with gr.Interface(fn=getTour, inputs=[
90
  gr.inputs.Textbox(label="Annual Household Income (in dollars)"),
 
93
  gr.inputs.Dropdown(["unknown", "male", "female"], label="Gender/sex"),
94
  gr.inputs.Dropdown(["unknown", "grade school","highschool", "associates", "bachelors", "graduate"], label="Educational attainment level"),
95
  gr.inputs.Dropdown(["unknown", "yes","no"], label="Worker status")],
96
+ outputs=["json",gr.HTML()], title="GPT-Tour", description="Generate a tour for a person", allow_flagging=False, allow_screenshot=False, allow_embedding=False) as iface:
 
97
  iface.launch()