Spaces:
Sleeping
Sleeping
File size: 4,205 Bytes
978fed5 | 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 | """Experiment workflow form and runner."""
from pathlib import Path
import streamlit as st
from loguru import logger
from utils import _rm_upload_root, find_data_analysis_file, save_and_extract_upload
from scider.workflows.experiment_workflow import ExperimentWorkflow
def run_experiment(q, path, workspace_path):
"""Run experiment workflow. Called from background thread."""
if path:
analysis_path = Path(path).resolve()
if not analysis_path.exists():
return f"Error: Analysis path does not exist: {analysis_path}", []
logger.info(f"Running experiment with analysis file: {analysis_path}")
w = ExperimentWorkflow.from_data_analysis_file(
workspace_path=workspace_path,
user_query=q,
data_analysis_path=str(analysis_path),
max_revisions=5,
recursion_limit=100,
)
else:
return "No data analysis file", []
w.run()
return w.final_summary or "Experiment finished", w.experiment_agent_intermediate_state
def render_form():
"""Render the experiment form. Returns workflow_config dict or None."""
with st.form("experiment_form", clear_on_submit=True):
st.markdown("### Experiment Workflow")
st.caption("Upload a zip containing data_analysis.md or enter path manually")
uploaded_exp_zip = st.file_uploader(
"Upload ZIP with data analysis (optional)",
type=["zip"],
key="exp_upload",
help="Zip containing data_analysis.md. Extracted temporarily, deleted on reset.",
)
if st.session_state.get("uploaded_experiment_path"):
st.info(f"Using: `{st.session_state.uploaded_experiment_path}`")
query = st.text_input("Experiment Query", placeholder="Describe your experiment...")
submitted = st.form_submit_button(
"Run Experiment",
)
if submitted:
if not query or not query.strip():
query = "Run an experiment based on the data analysis — build a model, evaluate it, and report results."
path_to_use = None
if uploaded_exp_zip:
prev = st.session_state.get("uploaded_experiment_path")
if prev:
_rm_upload_root(Path(prev))
if "uploaded_experiment_path" in st.session_state:
del st.session_state.uploaded_experiment_path
extracted = save_and_extract_upload(uploaded_exp_zip)
if extracted and extracted.exists():
extracted = extracted.resolve()
analysis_file = find_data_analysis_file(extracted)
if analysis_file and analysis_file.exists():
analysis_file = analysis_file.resolve()
st.session_state.uploaded_experiment_path = str(analysis_file)
st.session_state.workspace_path = analysis_file.parent
path_to_use = str(analysis_file)
st.success(f"Found analysis file: {path_to_use}")
else:
st.error(
f"Zip must contain data_analysis.md or analysis.md. Searched in: {extracted}"
)
else:
st.error(f"Failed to process uploaded zip file. Extracted path: {extracted}")
elif st.session_state.get("uploaded_experiment_path"):
p = Path(st.session_state.uploaded_experiment_path).resolve()
if p.exists():
path_to_use = str(p)
st.session_state.workspace_path = p.parent
else:
st.warning(f"Previously uploaded path no longer exists: {p}")
if "uploaded_experiment_path" in st.session_state:
del st.session_state.uploaded_experiment_path
if path_to_use:
return {"type": "experiment", "query": query, "path": path_to_use}
else:
st.error("Please upload a zip with data_analysis.md or enter a data analysis path.")
return None
|