| """Gradio demo for hv-reader.""" |
| import gradio as gr |
|
|
| from hv_reader import HVReader, SAMPLE_FICTION, SAMPLE_ACADEMIC, \ |
| SAMPLE_OVERCLAIM, SAMPLE_MONOTONE |
|
|
| model = HVReader() |
|
|
|
|
| def analyze_text(text: str) -> str: |
| if not text or not text.strip(): |
| return "Please enter some text." |
| return model.render(model.analyze(text)) |
|
|
|
|
| def profile_only(text: str) -> str: |
| if not text or not text.strip(): |
| return "Please enter some text." |
| r = model.analyze(text) |
| return ( |
| f"pace {r.pace:.3f} ({r.mean_wpm:.0f} WPM)\n" |
| f"memory {r.memory:.3f} (1-week retention)\n" |
| f"passes {r.passes:.3f} (reads needed)\n" |
| f"slip {r.slip:.3f} (attention lapse)\n" |
| f"wall {r.wall:.3f} (refusal)\n" |
| f"\n" |
| f"ttu {r.ttu_s:.1f} s\n" |
| f"hunger {r.hunger_final:+.2f}\n" |
| f"\n" |
| f"{r.summary}" |
| ) |
|
|
|
|
| with gr.Blocks(title="hv-reader") as demo: |
| gr.Markdown( |
| "# hv-reader\n" |
| "*The reading experience, in one call.*\n\n" |
| "Five-axis profile: pace, memory, passes, slip, wall." |
| ) |
| with gr.Row(): |
| with gr.Column(scale=2): |
| text_in = gr.Textbox( |
| label="Text", |
| lines=14, |
| placeholder="Paste a paragraph, essay, abstract...", |
| ) |
| with gr.Row(): |
| btn = gr.Button("Analyze", variant="primary") |
| p_btn = gr.Button("Profile only") |
| with gr.Column(scale=3): |
| out = gr.Textbox( |
| label="Report", |
| lines=34, |
| show_copy_button=True, |
| ) |
|
|
| btn.click(analyze_text, [text_in], out) |
| p_btn.click(profile_only, [text_in], out) |
|
|
| gr.Examples( |
| examples=[ |
| [SAMPLE_FICTION], |
| [SAMPLE_ACADEMIC], |
| [SAMPLE_OVERCLAIM], |
| [SAMPLE_MONOTONE], |
| ], |
| inputs=[text_in], |
| outputs=out, |
| fn=analyze_text, |
| ) |
|
|
|
|
| if __name__ == "__main__": |
| demo.launch() |
