debate / app.py
vikramvasudevan's picture
Upload folder using huggingface_hub
1a5dbae verified
from dotenv import load_dotenv
import gradio as gr
from src.debate.crew import Debate
PROPOSITION_DEFAULT = "### Arguments in favor\n\n_Arguments in favor will appear here_"
OPPOSITION_DEFAULT = "### Arguments against\n\n_Arguments against will appear here_"
DECISION_DEFAULT = "### The Judge's Decision\n\n_Decision will appear here_"
def debate_motion(motion):
debate = Debate()
crew = debate.crew()
inputs = {
'motion': motion
}
results = crew.kickoff(inputs=inputs)
print("Results: ", results)
proposition = debate.my_proposition_task.output.raw
opposition = debate.my_opposition_task.output.raw
decision = debate.my_judge_task.output.raw
yield proposition, opposition, decision
if __name__ == "__main__":
load_dotenv(override=True)
with gr.Blocks(theme=gr.themes.Soft()) as demo:
motion = gr.Text(
label="Motion",
value="Being Vegan is better for the environment",
submit_btn="Start Debate",
placeholder="Enter a motion to debate",
)
with gr.Column():
with gr.Row(equal_height=True):
proposition = gr.Markdown(
value=PROPOSITION_DEFAULT,
container=True,
)
opposition = gr.Markdown(
value=OPPOSITION_DEFAULT,
container=True,
)
decision = gr.Markdown(
value=DECISION_DEFAULT,
container=True,
)
motion.submit(
debate_motion, inputs=[motion], outputs=[proposition, opposition, decision]
)
demo.launch()