made app.py more readable

app.py

@@ -4,27 +4,12 @@ from structured_apparatus_chain import (
     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,
@@ -37,52 +22,36 @@ experiment_retriever_options = {
     "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
+    selected_chain = experiment_retriever_options[retriever_choice]
     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")],
+    inputs=["text", gr.Radio(choices=list(apparatus_retriever_options.keys()), label="Select a retriever", value="Wikipedia")],
     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(
+generate_experiment_interface = gr.Interface(
     fn=generate_experiment,
-    inputs=["text", gr.Radio(choices=list(experiment_retriever_options.keys()), label="Select a retriever")],
+    inputs=["text", gr.Radio(choices=list(experiment_retriever_options.keys()), label="Select a retriever", value="Wikipedia")],
     outputs="text",
     title="Generate an experiment",
-    description="I am here to generate a store science experiments for our users",
-    # examples=apparatus_examples,
+    description="I am here to generate and store science experiments for our users",
 )
 
-# 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"])
+    generate_experiment_interface,
+], ["Generate Apparatus", "Generate Experiment"])
 
 if __name__ == "__main__":
     demo.launch()