|
|
import gradio as gr |
|
|
import requests |
|
|
import pandas as pd |
|
|
|
|
|
def api_url(remote): |
|
|
return f"https://huynhdoo--mps-api-{remote}.modal.run" |
|
|
|
|
|
origins = { |
|
|
'Formation' : ['formation.presentation', 'formation.summary'], |
|
|
'Métier' : ['metier.presentation', 'metier.competences', |
|
|
'metier.condition_travail', 'metier.nature_travail', |
|
|
'metier.acces_metier', 'metier.vie_professionnelle', |
|
|
'metier.accroche_metier', 'metier.format_court1', |
|
|
'metier.format_court2'] |
|
|
} |
|
|
|
|
|
def retrieve(origin='Formation', query='cuisine', n_results=20): |
|
|
|
|
|
json = dict( |
|
|
query=query, |
|
|
origins=origins[origin], |
|
|
n_results=n_results |
|
|
) |
|
|
|
|
|
resp = requests.post(url=api_url('retrieve'), json=json) |
|
|
data = resp.json() |
|
|
|
|
|
|
|
|
distances = pd.DataFrame({'distance': data['distances']}) |
|
|
metadatas = pd.DataFrame(data['metadatas']) |
|
|
documents = pd.DataFrame({'document': data['documents']}) |
|
|
df = pd.concat([distances, metadatas, documents], axis=1) |
|
|
df['distance'] = df['distance'].apply(lambda x: round(x, 3)) |
|
|
df['origin'] = df['origin'].apply(lambda x: x.split('.')[1]) |
|
|
return df |
|
|
|
|
|
def rank(query='cuisine', documents = []): |
|
|
|
|
|
json = dict( |
|
|
query=query, |
|
|
documents=documents |
|
|
) |
|
|
|
|
|
try: |
|
|
resp = requests.post(url=api_url('rank'), json=json) |
|
|
data = resp.json() |
|
|
except: |
|
|
return [] |
|
|
|
|
|
|
|
|
return data['ranking'] |
|
|
|
|
|
gradio_app = gr.Interface( |
|
|
fn=retrieve, |
|
|
inputs=[ |
|
|
gr.Dropdown(choices=list(origins.keys()), value=list(origins.keys())[0], label="Origine", info="Choisir un type de donnée à interroger"), |
|
|
gr.Textbox(label="Recherche", info="Votre recherche"), |
|
|
gr.Number(value=10, label="Nombre de résultats", info="Nombre de résultats attendus") |
|
|
], |
|
|
outputs=[ |
|
|
gr.DataFrame(label="Résultats", headers=["Distance", "Key", "Label", "Origin", "Document"]) |
|
|
], |
|
|
examples=[['Formation', 'militaire'], |
|
|
['Métier', 'cuisine'], |
|
|
['Formation', 'écologie'], |
|
|
['Métier', 'eau'], |
|
|
['Formation', 'math'], |
|
|
], |
|
|
cache_examples=False |
|
|
) |
|
|
|
|
|
if __name__ == "__main__": |
|
|
gradio_app.launch(auth=("mps", "sup"), share=True) |
|
|
|
