Giving up on graph visualization

app.py

@@ -1,38 +1,21 @@
 import gradio as gr
 import json
-from pyvis.network import Network
 
 from transformers import pipeline
 # Load the pipeline
 tourModel = pipeline(model="manhan/GPT-Tour")
 
-def vizTour(activityList):
-    net = Network(directed=True,notebook=False)
-    net.add_node(activityList[0])
-    lastActivity = activityList[0]
-    for activity in activityList[1:]:
-        if activity not in net.get_nodes():
-            net.add_node(activity)
-        net.add_edge(lastActivity, activity)
-        lastActivity = activity
-    net.save_graph('tour.html')
-    #tour_html = open('tour.html', 'r')
-    tour_html = gr.HTML("<iframe src=\"tour.html\"></iframe>")
-    return tour_html
-
 def getTour(income,size,years,sex,edu,wrk): 
     # person = a dict with person-level and hh-level attributes:
     person = {}
     hh_income = int(income)
-    if hh_income<0:
-        _skip = 1 # nvm, bad data
-    elif hh_income<4: # $25,000
+    if hh_income<25000: # $25,000
         person['hh_inc'] = 'poor'
-    elif hh_income<6: # $50,000
+    elif hh_income<50000: # $50,000
         person['hh_inc'] = 'low'
-    elif hh_income<7: # $75,000
+    elif hh_income<75000: # $75,000
         person['hh_inc'] = 'medium'
-    elif hh_income<9: # $125,000
+    elif hh_income<125000: # $125,000
         person['hh_inc'] = 'high'
     else: # over
         person['hh_inc'] = 'affluent'
@@ -67,7 +50,6 @@ def getTour(income,size,years,sex,edu,wrk):
         end_pos = generated.find(']')+1
         activity_list_str = generated[start_pos:end_pos]
         print(f"Extracted: '{activity_list_str}'")
-        # this check doesn't appear to work anyways
         #if person['wrk']=='yes' and activity_list_str.find('Work')==-1:
         #    continue # try again
         #if person['wrk']=='no' and activity_list_str.find('Work')>0:
@@ -84,8 +66,7 @@ def getTour(income,size,years,sex,edu,wrk):
                 print(e)
         else:
             print("Nothing extracted!")
-    tour_html = vizTour(activity_list)
-    return activity_list, tour_html
+    return activity_list
 
 with gr.Interface(fn=getTour, inputs=[
     gr.inputs.Textbox(label="Annual Household Income (in dollars)"),
@@ -94,5 +75,5 @@ with gr.Interface(fn=getTour, inputs=[
     gr.inputs.Dropdown(["unknown", "male", "female"], label="Gender/sex"),
     gr.inputs.Dropdown(["unknown", "grade school","highschool", "associates", "bachelors", "graduate"], label="Educational attainment level"),
     gr.inputs.Dropdown(["unknown", "yes","no"], label="Worker status")], 
-    outputs=["json","html"], title="GPT-Tour", description="Generate a tour for a person", allow_flagging=False, allow_screenshot=False, allow_embedding=False) as iface:
+    outputs=["json"], title="GPT-Tour", description="Generate a tour for a person", allow_flagging='never') as iface:
     iface.launch()