Spaces:
Running
Running
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() | |