first commit

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import pandas as pd
+from transformers import pipeline
 
 # theme = gr.themes.Monochrome(spacing_size=gr.themes.sizes.spacing_md,
 #                              radius_size=gr.themes.sizes.radius_sm,
@@ -8,25 +9,32 @@ import pandas as pd
 #     slider_color="#FF0000"
 # )
 
-df = pd.read_json("description_subsector.json", orient='index')
-df = df.rename(columns={"old_name": "Subsector"})
-df.sort_values(by=['Subsector'], inplace=True)
+# df = pd.read_json("description_subsector.json", orient='index')
+# df = df.reset_index().rename(columns={'index': 'Subsector'})
+# df = df.sort_values(by=['Subsector'])
+# df.reset_index(drop=True, inplace=True)
+# df.to_csv('subsectors.csv', index=False)
+
+df = pd.read_csv('subsectors.csv')
 # df_column_width = [200, None, None]
 
+
 def click_button(model, abstract):
-    labels = {"Classificação 1": 0.75, "Classificação 2": 0.8, "Classificação 3": 0.2}
-    return labels
+    # labels = {"Classificação 1": 0.75, "Classificação 2": 0.8, "Classificação 3": 0.2}
+    classifier = pipeline("zero-shot-classification")
+    labels = df['Subsector'].tolist()
+    result = classifier(abstract, labels)
+    return {label: round(prob, 4) for label, prob in zip(result["labels"], result["scores"])}
+    # return labels
+
 
 def on_select(evt: gr.SelectData):  # SelectData is a subclass of EventData
-    selected = df.iloc[evt.index]
-    name = selected['Subsector'][0]
+    selected = df.iloc[[evt.index[0]]].iloc[0]
+    name, definition, keywords, does_include, does_not_include = selected['Subsector'], selected['Definition'], selected['Keywords'], selected['Does include'], selected['Does not include']
     name_accordion = gr.Accordion(label=name)
-    definition = selected['Definition'][0]
-    keywords = selected['Keywords'][0]
-    does_include = selected['Does include'][0]
-    does_not_include = selected['Does not include'][0]
     return name_accordion, definition, keywords, does_include, does_not_include
 
+
 # with gr.Blocks(theme=theme) as startup_genome_demo:
 with gr.Blocks() as startup_genome_demo:
     state_lotto = gr.State()
@@ -63,4 +71,4 @@ with gr.Blocks() as startup_genome_demo:
 
 if __name__ == "__main__":
     startup_genome_demo.queue()
-    startup_genome_demo.launch()
+    startup_genome_demo.launch()

requirements.txt

@@ -1,2 +1,3 @@
 gradio
-plotly
+plotly
+sentence-transformers