| import gradio as gr |
| import os |
| import json |
| from file_utils import ( |
| extract_text_from_file, |
| call_llm |
| ) |
|
|
|
|
| def file_dashboard(): |
| uploaded_files = gr.State([]) |
| file_sections = gr.State({}) |
| file_texts = gr.State({}) |
| chat_history = gr.State({}) |
| quiz_data = gr.State({}) |
| total_score = gr.State({}) |
|
|
| file_guide_list = gr.Radio(label="Your Study Guides", choices=[], interactive=True) |
| btn_file_view = gr.Button("📁 Files") |
|
|
| file_upload = gr.File(label="📤 Upload File", file_types=[".pdf", ".txt", ".md", ".pptx"]) |
| study_guide_output = gr.Markdown("Upload a file to generate a study guide.") |
| file_chat_output = gr.Row(visible=False) |
|
|
| with file_chat_output: |
| file_preview = gr.Textbox(label="📄 File Content Preview", lines=15, interactive=False) |
| chat_output = gr.Chatbot(label="💬 Chat with Document") |
|
|
| chat_input = gr.Textbox(label="Ask something about this file") |
| generate_quiz_btn = gr.Button("🧠 Generate Quiz from File") |
| quiz_container = gr.Column(visible=False) |
| quiz_score = gr.Textbox(label="Your Score", interactive=False, visible=False) |
|
|
| def store_file_and_generate_guide(file, current_files, section_map, text_map): |
| if file is None: |
| return current_files, section_map, text_map, "No file uploaded.", gr.update(choices=[]) |
| text = extract_text_from_file(file) |
| filename = os.path.basename(file.name) |
| prompt = f"Break the content into 4–6 logical learning sections. Format: 'Section Title: Short summary'\n\n{text}" |
| response = call_llm(prompt, system_message="You are a helpful assistant for students studying documents.") |
| 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, f"### {filename} Study Guide\n\n{response}", gr.update(choices=updated_files) |
|
|
| def load_study_guide(selected_filename, section_map): |
| if not selected_filename or selected_filename not in section_map: |
| return "No study guide available." |
| return f"### {selected_filename} Study Guide\n\n{section_map[selected_filename]}" |
|
|
| def open_file_viewer(selected_file, text_map): |
| content = text_map.get(selected_file, "No file content available.") |
| return gr.update(visible=True), content, [] |
|
|
| def chat_with_file(question, selected_file, text_map, history_map): |
| if not selected_file or selected_file not in text_map: |
| return [], history_map |
| text = text_map[selected_file] |
| history = history_map.get(selected_file, []) |
| prompt = f"You are a helpful tutor. Answer this question using the document content below.\n\n{text}\n\nQuestion: {question}" |
| answer = call_llm(prompt) |
| new_history = history + [(question, answer)] |
| history_map[selected_file] = new_history |
| return new_history, history_map |
|
|
| def generate_quiz_ui(selected_file, text_map): |
| if not selected_file or selected_file not in text_map: |
| return gr.update(visible=True), gr.update(visible=False, value="") |
|
|
| text = text_map[selected_file] |
| prompt = f"Create 5 multiple-choice questions based on the content below.\nReturn as JSON list with 'question', 'options' (list), and 'answer'.\n\n{text}" |
| response = call_llm(prompt) |
|
|
| try: |
| questions = json.loads(response) |
| quiz_elements = [] |
| score = {"correct": 0, "total": len(questions)} |
|
|
| for i, q in enumerate(questions): |
| question_md = gr.Markdown(f"**Q{i+1}. {q['question']}**") |
| options = gr.Radio(choices=q['options'], label=f"Your Answer to Q{i+1}", interactive=True) |
| feedback = gr.Textbox(label="Result", interactive=False) |
|
|
| def evaluate(ans): |
| def check(choice): |
| correct = choice == ans |
| return "✅ Correct!" if correct else f"❌ Incorrect. Correct answer: {ans}" |
| return check |
|
|
| options.change(fn=evaluate(q['answer']), inputs=[options], outputs=[feedback]) |
| quiz_elements.extend([question_md, options, feedback]) |
|
|
| return gr.update(visible=True), gr.update(visible=True, value=f"Score: 0/{score['total']}") |
|
|
| except Exception as e: |
| return gr.update(visible=True), gr.update(visible=True, value="Error generating quiz.") |
|
|
| file_upload.change( |
| fn=store_file_and_generate_guide, |
| inputs=[file_upload, uploaded_files, file_sections, file_texts], |
| outputs=[uploaded_files, file_sections, file_texts, study_guide_output, file_guide_list], |
| show_progress="minimal" |
| ) |
|
|
| file_guide_list.change( |
| fn=load_study_guide, |
| inputs=[file_guide_list, file_sections], |
| outputs=[study_guide_output], |
| show_progress="minimal" |
| ) |
|
|
| btn_file_view.click( |
| fn=open_file_viewer, |
| inputs=[file_guide_list, file_texts], |
| outputs=[file_chat_output, file_preview, chat_output] |
| ) |
|
|
| chat_input.submit( |
| fn=chat_with_file, |
| inputs=[chat_input, file_guide_list, file_texts, chat_history], |
| outputs=[chat_output, chat_history], |
| show_progress="minimal" |
| ) |
|
|
| generate_quiz_btn.click( |
| fn=generate_quiz_ui, |
| inputs=[file_guide_list, file_texts], |
| outputs=[quiz_container, quiz_score], |
| show_progress="minimal" |
| ) |
|
|
| return gr.Column( |
| components=[ |
| file_upload, |
| study_guide_output, |
| file_chat_output, |
| chat_input, |
| generate_quiz_btn, |
| quiz_container, |
| quiz_score |
| ], |
| scale=4, |
| elem_id="main-column" |
| ) |
