Update file_viewer_panel.py

file_viewer_panel.py

@@ -1,136 +1,78 @@
 import gradio as gr
 import os
-import shutil
 import json
-import tempfile
-import uuid
-from pdf2image import convert_from_path
-from file_utils import extract_text_from_file, call_llm, save_user_file, fetch_user_files
-
-def make_public_file(img_path):
-    public_dir = "/tmp/gradio-public"
-    os.makedirs(public_dir, exist_ok=True)
-    filename = f"{uuid.uuid4()}.png"
-    public_path = os.path.join(public_dir, filename)
-    shutil.copy(img_path, public_path)
-    return public_path
-
-def file_viewer_dashboard(nickname):
+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:
-        with gr.Row():
-            with gr.Column(scale=3):
-                file_upload = gr.File(label="📂 Upload a File", file_types=[".pdf", ".txt", ".md", ".pptx"])
-                image_viewer = gr.HTML(label="📄 PDF Viewer")
-                load_more_btn = gr.Button("➕ Load More Pages", visible=False)
-
-            with gr.Column(scale=2):
-                generate_quiz_btn = gr.Button("🧠 Practice Questions")
-                quiz_container = gr.Column(visible=False)
-                chat_output = gr.Chatbot(label="Ask a question about the file")
-                chat_input = gr.Textbox(label="", placeholder="Ask ChatEDU a question...")
+        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_state = gr.State([])
-        all_image_paths = gr.State([])
-        loaded_page_count = gr.State(0)
 
-        def prepare_pdf_images(file):
-            if not file or not nickname:
-                return "", [], 0, gr.update(visible=False)
+        chat_input = gr.Textbox(label="🔍 Ask about the file")
+        chat_output = gr.Chatbot()
+
+        btn_accessible = gr.Button("♿️ Improve Accessibility")
+        accessible_box = gr.Textbox(label="🌍 Easier-to-Read Version", lines=15, interactive=False)
 
-            filename = os.path.basename(file.name)
-            saved_path = f"/tmp/{filename}"
-            shutil.copy(file.name, saved_path)
-            save_user_file(nickname.value, filename, saved_path)
+        quiz_box = gr.Column(visible=False)
+        quiz_score = gr.Textbox(label="Your Score", interactive=False, 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):
+            prompt = f"Use this document to answer:
+
+{text}
+
+Question: {query}"
+            return [(query, call_llm(prompt))]
 
-            ext = os.path.splitext(filename)[1].lower()
-            if ext != ".pdf":
-                html = f"<pre style='padding:12px; background:#f9f9f9; white-space:pre-wrap;'>{text}</pre>"
-                return html, [], 0, gr.update(visible=False)
-
-            temp_dir = tempfile.mkdtemp()
-            images = convert_from_path(saved_path, dpi=150, fmt="png", output_folder=temp_dir)
-
-            paths = []
-            for i, img in enumerate(images):
-                img_path = os.path.join(temp_dir, f"page_{i}.png")
-                img.save(img_path, "PNG")
-                public_path = make_public_file(img_path)
-                paths.append(public_path)
-
-            initial_html = "".join([
-                f"<img src='file={paths[i]}' style='width:100%; margin-bottom:12px; border-radius:8px;'/>"
-                for i in range(min(3, len(paths)))
-            ])
-            return initial_html, paths, min(3, len(paths)), gr.update(visible=(len(paths) > 3))
-
-        def load_more_pages(paths, current_count):
-            next_count = current_count + 3
-            total = len(paths)
-
-            new_html = "".join([
-                f"<img src='file={paths[i]}' style='width:100%; margin-bottom:12px; border-radius:8px;'/>"
-                for i in range(current_count, min(next_count, total))
-            ])
-
-            show_more = next_count < total
-            return gr.update(value=new_html, append=True), next_count, gr.update(visible=show_more)
-
-        def ask_question(user_input, text, history):
-            if not user_input.strip():
-                return history, history
-            messages = [
-                {"role": "system", "content": "You are a tutor. Only use the document below."},
-                {"role": "user", "content": f"Document:\n{text}"}
-            ]
-            for user_msg, bot_reply in history:
-                messages.append({"role": "user", "content": user_msg})
-                messages.append({"role": "assistant", "content": bot_reply})
-            messages.append({"role": "user", "content": user_input})
-            response = call_llm(messages=messages)
-            history.append((user_input, response))
-            return history, history
+        def make_accessible_version(text):
+            prompt = f"""
+You are an expert in reading accessibility.
+
+Reformat the following content to improve readability and make it easier to follow:
+- Use simple language
+- Break long paragraphs
+- Use bullet points or headings when appropriate
+- Preserve the meaning clearly
+
+Text:
+{text}
+"""
+            return call_llm(prompt)
 
         def generate_quiz(text):
-            prompt = (
-                "Generate 5 multiple choice questions from the document. "
-                "Each question should have:\n- question\n- options (list)\n- answer (string). Return JSON."
-            )
+            prompt = "Create 5 multiple-choice questions from this text. Return JSON list."
+            response = call_llm(prompt)
             try:
-                result = call_llm(prompt=prompt)
-                questions = json.loads(result)
-                components = []
-                for idx, q in enumerate(questions):
-                    q_md = gr.Markdown(f"**Q{idx+1}. {q['question']}**")
-                    radio = gr.Radio(choices=q["options"], label="Choose an answer")
-                    feedback = gr.Textbox(interactive=False)
-
-                    def check(ans, correct=q["answer"]):
-                        return "✅ Correct!" if ans == correct else f"❌ Correct answer: {correct}"
-
-                    radio.change(fn=check, inputs=[radio], outputs=[feedback])
-                    components.extend([q_md, radio, feedback])
-                return gr.update(visible=True), components
-            except:
-                return gr.update(visible=False), []
+                questions = json.loads(response)
+                elements = []
 
-        file_upload.change(
-            fn=prepare_pdf_images,
-            inputs=[file_upload],
-            outputs=[image_viewer, all_image_paths, loaded_page_count, load_more_btn]
-        )
+                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)
 
-        load_more_btn.click(
-            fn=load_more_pages,
-            inputs=[all_image_paths, loaded_page_count],
-            outputs=[image_viewer, loaded_page_count, load_more_btn]
-        )
+                    def grade(choice, correct=q["answer"]):
+                        return "✅ Correct!" if choice == correct else f"❌ Correct: {correct}"
 
-        chat_input.submit(fn=ask_question, inputs=[chat_input, file_text_state, chat_state],
-                          outputs=[chat_output, chat_state])
+                    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.")
 
-        generate_quiz_btn.click(fn=generate_quiz, inputs=[file_text_state], outputs=[quiz_container])
+        file_upload.change(fn=load_and_preview, inputs=[file_upload], outputs=[preview, file_text_state])
+        chat_input.submit(fn=handle_chat, inputs=[chat_input, file_text_state], outputs=[chat_output])
+        btn_accessible.click(fn=make_accessible_version, inputs=[file_text_state], outputs=[accessible_box])
+        gr.Button("🧠 Generate Quiz").click(fn=generate_quiz, inputs=[file_text_state], outputs=[quiz_box, quiz_box, quiz_score])
 
     return file_panel