| import gradio as gr |
| import os |
| import json |
| from file_utils import extract_text_from_file, call_llm |
|
|
| def file_viewer_dashboard(nickname_input): |
| with gr.Column(scale=4, elem_id="main-column") as file_panel: |
| file_upload = gr.File(label="📂 Upload a File", file_types=[".pdf", ".txt", ".md", ".pptx"]) |
| preview = gr.Textbox(label="📄 File Content Preview", lines=10, interactive=False) |
|
|
| file_text_state = gr.State("") |
| chat_history = gr.State([]) |
|
|
| chat_output = gr.Chatbot() |
|
|
| chat_input = gr.Textbox( |
| placeholder="Ask about the file... (press Enter or click 📤)", |
| label=None, |
| lines=1, |
| elem_id="chat-input", |
| ) |
| |
| send_btn = gr.Button("📤", size="sm", elem_id="chat-send-btn") |
|
|
| btn_simplify = gr.Button("🧾 Simplify Content") |
| simplified_box = gr.Textbox(label="📘 Simplified File Version", lines=15, interactive=False) |
|
|
| quiz_btn = gr.Button("🧠 Generate Quiz") |
| quiz_score = gr.Textbox(label="Your Score", interactive=False, visible=False) |
| quiz_box = gr.Column(visible=False) |
|
|
| def load_and_preview(file): |
| if not file: |
| return "", "" |
| text = extract_text_from_file(file) |
| return text, text |
|
|
| def handle_chat(query, text, history): |
| prompt = f"Refer only to this file and answer:\n\n{text}\n\nQuestion: {query}" |
| answer = call_llm(prompt) |
| history.append((query, answer)) |
| return history, "" |
|
|
| def simplify_content(text): |
| prompt = f""" |
| Reformat the following content to improve readability and clarity: |
| - Use simple language |
| - Shorter paragraphs |
| - Larger font layout (for display) |
| - Use clear sections or bullet points when needed |
| - Preserve all original meaning |
| |
| Text: |
| {text} |
| """ |
| return call_llm(prompt) |
|
|
| def generate_quiz(text): |
| prompt = "Create 5 multiple-choice questions from this text. Return JSON list." |
| response = call_llm(prompt) |
| try: |
| questions = json.loads(response) |
| elements = [] |
|
|
| for i, q in enumerate(questions): |
| q_md = gr.Markdown(f"**Q{i+1}. {q['question']}**") |
| q_radio = gr.Radio(q["options"], label="Select an answer") |
| q_feedback = gr.Textbox(label="Result", interactive=False) |
|
|
| def grade(choice, correct=q["answer"]): |
| return "✅ Correct!" if choice == correct else f"❌ Correct: {correct}" |
|
|
| q_radio.change(fn=grade, inputs=[q_radio], outputs=[q_feedback]) |
| elements.extend([q_md, q_radio, q_feedback]) |
|
|
| return gr.update(visible=True), elements, gr.update(visible=True, value="Score: 0/5") |
| except: |
| return gr.update(visible=False), [], gr.update(visible=True, value="Error generating quiz.") |
|
|
| file_upload.change(fn=load_and_preview, inputs=[file_upload], outputs=[preview, file_text_state]) |
| send_btn.click(fn=handle_chat, inputs=[chat_input, file_text_state, chat_history], outputs=[chat_output, chat_input]) |
| chat_input.submit(fn=handle_chat, inputs=[chat_input, file_text_state, chat_history], outputs=[chat_output, chat_input]) |
| btn_simplify.click(fn=simplify_content, inputs=[file_text_state], outputs=[simplified_box]) |
| quiz_btn.click(fn=generate_quiz, inputs=[file_text_state], outputs=[quiz_box, quiz_box, quiz_score]) |
|
|
| return file_panel |
