| import gradio as gr |
| import os |
| import json |
| from file_utils import ( |
| extract_text_from_file, |
| call_llm, |
| save_study_guide, |
| fetch_study_guides |
| ) |
|
|
| def file_study_guide_dashboard(nickname): |
| uploaded_files = gr.State([]) |
| file_sections = gr.State({}) |
| file_texts = gr.State({}) |
|
|
| with gr.Column(scale=4, elem_id="main-column") as guide_content: |
| nickname_input = nickname |
| file_upload = gr.File(label="📤 Upload File for Study Guide", file_types=[".pdf", ".txt", ".md", ".pptx"]) |
| study_guide_output = gr.Textbox(label="📘 Generated Study Guide", lines=10, interactive=False) |
| save_guide_btn = gr.Button("📌 Save Study Guide") |
|
|
| saved_guides_dropdown = gr.Dropdown(label="📚 Your Saved Guides", choices=[], interactive=True) |
| saved_guide_display = gr.Textbox(label="📖 Selected Guide Content", lines=10, interactive=False) |
|
|
| def generate_study_guide(file, current_files, section_map, text_map): |
| if file is None: |
| return current_files, section_map, text_map, "No file uploaded." |
|
|
| text = extract_text_from_file(file) |
| filename = os.path.basename(file.name) |
|
|
| prompt = f"Divide this document into 4–6 learning sections. For each section, include a bold title and 2–3 line explanation.\n\n{text}" |
| response = call_llm(prompt, system_message="You are an AI tutor creating study guides.") |
|
|
| updated_files = current_files + [filename] if filename not in current_files else current_files |
| section_map[filename] = response |
| text_map[filename] = text |
|
|
| return updated_files, section_map, text_map, response |
|
|
| def save_guide_to_db(user, uploaded_files, section_map): |
| if not uploaded_files or user.strip() == "": |
| return |
| filename = uploaded_files[-1] |
| guide = section_map.get(filename, "") |
| if guide: |
| save_study_guide(user=user, filename=filename, guide=guide) |
|
|
| def load_user_guides(user): |
| guides = fetch_study_guides(user) |
| return [f"{g.filename} ({g.timestamp})" for g in guides] |
|
|
| def show_saved_guide(user, label): |
| guides = fetch_study_guides(user) |
| for g in guides: |
| if label.startswith(g.filename): |
| return g.guide |
| return "" |
|
|
| file_upload.change( |
| fn=generate_study_guide, |
| inputs=[file_upload, uploaded_files, file_sections, file_texts], |
| outputs=[uploaded_files, file_sections, file_texts, study_guide_output] |
| ) |
|
|
| save_guide_btn.click( |
| fn=save_guide_to_db, |
| inputs=[nickname_input, uploaded_files, file_sections], |
| outputs=[] |
| ) |
|
|
| nickname_input.change( |
| fn=load_user_guides, |
| inputs=[nickname_input], |
| outputs=[saved_guides_dropdown] |
| ) |
|
|
| saved_guides_dropdown.change( |
| fn=show_saved_guide, |
| inputs=[nickname_input, saved_guides_dropdown], |
| outputs=[saved_guide_display] |
| ) |
|
|
| return guide_content |
|
|
