Update file_viewer_panel.py

file_viewer_panel.py

@@ -5,24 +5,23 @@ 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():
-            file_upload = gr.File(label="📂 Upload a File", file_types=[".pdf", ".txt", ".md", ".pptx"])
-
+        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([])
 
-        with gr.Row():
-            chat_input = gr.Textbox(label="🔍 Ask about the file")
-            chat_output = gr.Chatbot()
+        chat_input = gr.Textbox(label="🔍 Ask about the file")
+        chat_output = gr.Chatbot()
 
-        with gr.Row():
-            btn_accessible = gr.Button("♿️ Improve Accessibility")
-            accessible_box = gr.Textbox(label="🌍 Easier-to-Read Version", lines=15, interactive=False)
+        btn_accessible = gr.Button("♿️ Improve Accessibility")
+        accessible_box = gr.Textbox(label="🌍 Easier-to-Read Version", lines=15, interactive=False)
 
-        with gr.Row():
-            quiz_btn = gr.Button("🧠 Generate Quiz")
-            quiz_score = gr.Textbox(label="Your Score", interactive=False, visible=False)
+        btn_readability = gr.Button("🧾 Improve Readability", size="sm")
+        readable_box = gr.Textbox(label="📘 Reformatted 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):
@@ -31,9 +30,11 @@ def file_viewer_dashboard(nickname_input):
             text = extract_text_from_file(file)
             return text, text
 
-        def handle_chat(query, text):
-            prompt = f"Use this document to answer:\n\n{text}\n\nQuestion: {query}"
-            return [(query, call_llm(prompt))]
+        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 make_accessible_version(text):
             prompt = f"""
@@ -44,6 +45,19 @@ Reformat the following content to improve readability and make it easier to foll
 - Use bullet points or headings when appropriate
 - Preserve the meaning clearly
 
+Text:
+{text}
+"""
+            return call_llm(prompt)
+
+        def improve_readability(text):
+            prompt = f"""
+Please rewrite the following content to improve readability:
+- Increase spacing
+- Use larger font and clearer structure (for display purposes)
+- Break into shorter paragraphs or bullet points
+- Keep the meaning intact
+
 Text:
 {text}
 """
@@ -72,8 +86,9 @@ Text:
                 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])
-        chat_input.submit(fn=handle_chat, inputs=[chat_input, file_text_state], outputs=[chat_output])
+        chat_input.submit(fn=handle_chat, inputs=[chat_input, file_text_state, chat_history], outputs=[chat_output, chat_input])
         btn_accessible.click(fn=make_accessible_version, inputs=[file_text_state], outputs=[accessible_box])
+        btn_readability.click(fn=improve_readability, inputs=[file_text_state], outputs=[readable_box])
         quiz_btn.click(fn=generate_quiz, inputs=[file_text_state], outputs=[quiz_box, quiz_box, quiz_score])
 
-    return file_panel
+    return file_panel