events / app.py
cryptocalypse's picture
Update app.py
d66d33e verified
import gradio as gr
import gen
import psychohistory
with gr.Blocks(title="PSYCHOHISTORY") as app:
with gr.Tab("Search"):
with gr.Row():
model = gr.Radio(["Gemma2", "Zephy7B"], value="Gemma2",label="Model", info="Select model",scale=2)
time_direction = gr.Radio(["Past", "Future"],value="Future", label="Direction", info="Select time direction",scale=2)
num_fut = gr.Textbox(value="4", label="Future Steps", scale=2)
with gr.Column():
spaces_include = gr.Checkbox(label="Hight", value=True,info="Probabilities")
spaces_include = gr.Checkbox(label="Medium", value=True)
spaces_include = gr.Checkbox(label="Low", value=True)
with gr.Row():
txt_search = gr.Textbox(value="Peace in Eurasia in 2027", label="Search Term", scale=5)
btn_search = gr.Button("Look", scale=1)
with gr.Row():
img_output = gr.Image(label="Global Graph ", type="filepath") # Add an Image component
img_output2 = gr.Image(label="Most likely", type="filepath")
img_output3 = gr.Image(label="The slowest", type="filepath")
# Trigger graph generation after JSON is generated
with gr.Row():
mem_results = gr.JSON(label="Predicted Futures",scale=2) # Futures predicted
mem_results2 = gr.JSON(label="Radians Slow Path",scale=1)
mem_results3 = gr.JSON(label="Radians Best Path",scale=1)
#counter_results = gr.JSON(label="Counter Activity") # Wich activities are against the event
#dark_results = gr.JSON(label="Dark Forces") #Who makes not happend
btn_search.click(
gen.generate,
inputs=[txt_search],
outputs=mem_results
)
mem_results.change(
psychohistory.main,
inputs=[mem_results],
outputs=[img_output,img_output2,img_output3]#,mem_results2,mem_results3]
)
if __name__ == "__main__":
app.launch()