Spaces:
Runtime error
Runtime error
File size: 2,785 Bytes
b49a392 3795f1b b49a392 3795f1b 7ce3f04 3795f1b b49a392 3795f1b 734a0bd 3795f1b 734a0bd 3795f1b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 |
import gradio as gr
from structured_apparatus_chain import (
arxiv_chain as apparatus_arxiv_chain,
pub_med_chain as apparatus_pub_med_chain,
wikipedia_chain as apparatus_wikipedia_chain
)
from structured_experiment_chain import (
arxiv_chain as experiment_arxiv_chain,
pub_med_chain as experiment_pub_med_chain,
wikipedia_chain as experiment_wikipedia_chain
)
# apparatus_examples = [
# "Microscope",
# "I don't fucking know yet",
# ]
# apparatus_examples = {
# ["sample": "cool"],
# ["guitar": "john"],
# }
# apparatus_examples = [apparatus_examples]
apparatus_retriever_options = {
"Arxiv": apparatus_arxiv_chain,
"PubMed": apparatus_pub_med_chain,
"Wikipedia": apparatus_wikipedia_chain,
}
experiment_retriever_options = {
"Arxiv": experiment_arxiv_chain,
"PubMed": experiment_pub_med_chain,
"Wikipedia": experiment_wikipedia_chain,
}
# experiment_retriever_options = ["Option 1", "Option 2", "Option 3"]
def generate_apparatus(input_text, retriever_choice):
# Retrieve the appropriate chain based on the selected option
selected_chain = apparatus_retriever_options[retriever_choice]
# Execute the selected chain with the input text
output_text = selected_chain.invoke(input_text)
return output_text
def generate_experiment(input_text, retriever_choice):
# Retrieve the appropriate chain based on the selected option
selected_chain = apparatus_retriever_options[retriever_choice]
# Execute the selected chain with the input text
output_text = selected_chain.invoke(input_text)
return output_text
generate_apparatus_interface = gr.Interface(
fn=generate_apparatus,
inputs=["text", gr.Radio(choices=list(apparatus_retriever_options.keys()), label="Select a retriever")],
outputs="text",
title="Generate Apparatus",
description="I am here to help makers make more and learn the science behind things",
# examples=apparatus_examples,
)
experiment_apparatus_interface = gr.Interface(
fn=generate_experiment,
inputs=["text", gr.Radio(choices=list(experiment_retriever_options.keys()), label="Select a retriever")],
outputs="text",
title="Generate an experiment",
description="I am here to generate a store science experiments for our users",
# examples=apparatus_examples,
)
# stt_demo = gr.Interface(
# fn="huggingface/facebook/wav2vec2-base-960h",
# inputs=gr.Microphone(type="filepath"),
# outputs=None,
# title="Generate Experiment",
# description="Let me try to guess what you're saying!"
# )
demo = gr.TabbedInterface([
generate_apparatus_interface,
experiment_apparatus_interface,
], ["Generate Apparatus", "Generate Experiment"])
if __name__ == "__main__":
demo.launch()
|