Update app.py

app.py

@@ -99,34 +99,25 @@ def markdown_to_story(markdown_text: str, font_name: str, emoji_font: str):
     style_h1 = ParagraphStyle('h1', fontName=bold_font, spaceBefore=12, fontSize=24, textColor=colors.HexColor("#1E3A8A"))
     style_h2 = ParagraphStyle('h2', fontName=bold_font, spaceBefore=10, fontSize=18, textColor=colors.HexColor("#374151"))
     style_h3 = ParagraphStyle('h3', fontName=bold_font, spaceBefore=8, fontSize=14, textColor=colors.HexColor("#4B5563"))
-    style_code = ParagraphStyle('Code', fontName='Courier', backColor=colors.whitesmoke, textColor=colors.darkred, borderWidth=1, borderColor=colors.lightgrey, padding=8)
-
+    
     story, first_heading = [], True
     for line in markdown_text.split('\n'):
         stripped_line = line.strip()
         if not stripped_line:
             story.append(Spacer(1, 0.1 * inch)); continue
 
-        # Determine the structural element and its style
         content, style, extra_args = stripped_line, style_normal, {}
         if stripped_line.startswith("# "):
             if not first_heading: story.append(PageBreak())
             content, style, first_heading = stripped_line.lstrip('# '), style_h1, False
-        elif stripped_line.startswith("## "):
-            content, style = stripped_line.lstrip('## '), style_h2
-        elif stripped_line.startswith("### "):
-            content, style = stripped_line.lstrip('### '), style_h3
-        elif stripped_line.startswith(("- ", "* ")):
-            content, extra_args = stripped_line[2:], {'bulletText': '•'}
+        elif stripped_line.startswith("## "): content, style = stripped_line.lstrip('## '), style_h2
+        elif stripped_line.startswith("### "): content, style = stripped_line.lstrip('### '), style_h3
+        elif stripped_line.startswith(("- ", "* ")): content, extra_args = stripped_line[2:], {'bulletText': '•'}
         
-        # Now, format the content string correctly for ReportLab
-        # Apply bold/italic first
         formatted_content = re.sub(r'_(.*?)_', r'<i>\1</i>', re.sub(r'\*\*(.*?)\*\*', r'<b>\1</b>', content))
-        # Then, apply the emoji font tags. This order is crucial.
         final_content = apply_emoji_font(formatted_content, emoji_font)
         
         story.append(Paragraph(final_content, style, **extra_args))
-        
     return story
 
 
@@ -193,13 +184,22 @@ def build_file_explorer_html(generated_files, pdf_files_for_gallery):
     
     file_explorer_html = ""
     file_icons = {".pdf": "📄", ".docx": "📝", ".xlsx": "📊"}
+    action_buttons = """
+        <div class="actions">
+            <button class="action-btn" onclick="event.preventDefault(); alert('QuizMe: Feature coming soon!')">🧠 QuizMe</button>
+            <button class="action-btn" onclick="event.preventDefault(); alert('Revise: Feature coming soon!')">✍️ Revise</button>
+        </div>
+    """
     for file_path in generated_files:
         icon = file_icons.get(file_path.suffix, '📎')
         file_explorer_html += f"""
-        <a href="/file={file_path}" class="file-link" download="{file_path.name}">
-            <span class="file-icon">{icon}</span>
-            <span class="file-name">{file_path.name}</span>
-        </a>
+        <div class="file-item">
+            <a href="/file={file_path}" class="file-link" download="{file_path.name}">
+                <span class="file-icon">{icon}</span>
+                <span class="file-name">{file_path.name}</span>
+            </a>
+            {action_buttons if file_path.suffix == '.pdf' else ''}
+        </div>
         """
 
     gallery_items = []
@@ -208,103 +208,79 @@ def build_file_explorer_html(generated_files, pdf_files_for_gallery):
         if preview_path:
             with open(preview_path, "rb") as f:
                 img_base64 = base64.b64encode(f.read()).decode("utf-8")
-                gallery_items.append({
-                    "preview_src": f"data:image/png;base64,{img_base64}",
-                    "filename": pdf_path.name
-                })
+                gallery_items.append({"preview_src": f"data:image/png;base64,{img_base64}", "filename": pdf_path.name})
     
     gallery_html = ""
     if gallery_items:
-        thumbs_html = ""
-        for item in gallery_items:
-            thumbs_html += f'<img src="{item["preview_src"]}" class="thumbnail" onclick="selectThumbnail(this, \'{item["preview_src"]}\', \'{item["filename"]}\')">'
-        
+        thumbs_html = "".join([f'<img src="{item["preview_src"]}" class="thumbnail" onclick="selectThumbnail(this, \'{item["preview_src"]}\', \'{item["filename"]}\')">' for item in gallery_items])
         gallery_html = f"""
         <div class="gallery-container">
-            <div class="main-view">
-                <img id="main-image" src="{gallery_items[0]['preview_src']}" class="main-image">
-                <p id="main-filename">{gallery_items[0]['filename']}</p>
-            </div>
+            <div class="main-view"><img id="main-image" src="{gallery_items[0]['preview_src']}" class="main-image"></div>
             <div class="thumbnail-strip">{thumbs_html}</div>
         </div>
+        <p id="main-filename">{gallery_items[0]['filename']}</p>
         """
     
     html = f"""
     <style>
         .tabs {{ display: flex; border-bottom: 2px solid #ccc; }}
-        .tab-button {{ padding: 10px 15px; cursor: pointer; background: #f1f1f1; border: none; border-bottom: 2px solid transparent; outline: none; }}
-        .tab-button.active {{ background: #fff; border-top: 2px solid #007bff; border-left: 2px solid #ccc; border-right: 2px solid #ccc; border-bottom: 2px solid #fff; }}
+        .tab-button {{ padding: 10px 15px; cursor: pointer; background: #f1f1f1; border: none; font-size: 1.2em; }}
+        .tab-button.active {{ background: #fff; border-top: 2px solid #6366f1; border-left: 1px solid #ccc; border-right: 1px solid #ccc; position: relative; top: 1px;}}
         .tab-content {{ display: none; padding: 15px; border: 1px solid #ccc; border-top: none; }}
         .tab-content.active {{ display: block; }}
-        .file-explorer {{ display: grid; grid-template-columns: repeat(auto-fill, minmax(200px, 1fr)); gap: 10px; }}
-        .file-link {{ display: flex; align-items: center; padding: 10px; background: #f9f9f9; border-radius: 5px; text-decoration: none; color: #333; }}
-        .file-link:hover {{ background: #e9e9e9; }}
-        .file-icon {{ font-size: 2.5em; margin-right: 10px; }}
+        .file-explorer {{ display: grid; grid-template-columns: repeat(auto-fill, minmax(250px, 1fr)); gap: 15px; }}
+        .file-item {{ background: #f9f9f9; border-radius: 8px; box-shadow: 0 2px 4px rgba(0,0,0,0.1); }}
+        .file-link {{ display: flex; align-items: center; padding: 15px; text-decoration: none; color: #333; }}
+        .file-icon {{ font-size: 3em; margin-right: 15px; }}
+        .file-name {{ font-weight: bold; word-break: break-all; }}
+        .actions {{ display: flex; justify-content: space-around; padding: 5px 10px; border-top: 1px solid #eee; }}
+        .action-btn {{ background: none; border: none; cursor: pointer; font-size: 1.1em; }}
         .gallery-container {{ display: flex; height: 500px; }}
-        .main-view {{ flex: 3; padding: 10px; display: flex; flex-direction: column; align-items: center; justify-content: center; }}
-        .main-image {{ max-width: 100%; max-height: 90%; }}
-        .thumbnail-strip {{ flex: 1; overflow-y: auto; padding: 5px; }}
-        .thumbnail {{ width: 100%; margin-bottom: 5px; cursor: pointer; border: 2px solid transparent; }}
-        .thumbnail.active {{ border-color: #007bff; }}
+        .main-view {{ flex: 4; }} .main-image {{ width: 100%; height: 100%; object-fit: contain; }}
+        .thumbnail-strip {{ flex: 1; overflow-y: auto; }} .thumbnail {{ width: 100%; margin-bottom: 5px; cursor: pointer; border: 3px solid transparent; }}
+        .thumbnail.active {{ border-color: #6366f1; }}
+        #main-filename {{ text-align: center; font-weight: bold; margin-top: 10px; }}
     </style>
     <div class="tabs">
-        <button class="tab-button active" onclick="openTab(event, 'explorer')">🗂️ File Explorer</button>
-        {'<button class="tab-button" onclick="openTab(event, \'gallery\')">🖼️ PDF Gallery</button>' if gallery_items else ''}
-    </div>
-    <div id="explorer" class="tab-content active">
-        <div class="file-explorer">{file_explorer_html}</div>
-    </div>
-    <div id="gallery" class="tab-content">
-        {gallery_html}
+        <button class="tab-button active" onclick="openTab(event, 'explorer')">🗂️ FileExplorer</button>
+        {'<button class="tab-button" onclick="openTab(event, \'gallery\')">🖼️ GlimpsePDFs</button>' if gallery_items else ''}
     </div>
+    <div id="explorer" class="tab-content active"><div class="file-explorer">{file_explorer_html}</div></div>
+    <div id="gallery" class="tab-content">{gallery_html}</div>
     <script>
         function openTab(evt, tabName) {{
-            var i, tabcontent, tablinks;
-            tabcontent = document.getElementsByClassName("tab-content");
-            for (i = 0; i < tabcontent.length; i++) {{ tabcontent[i].style.display = "none"; }}
-            tablinks = document.getElementsByClassName("tab-button");
-            for (i = 0; i < tablinks.length; i++) {{ tablinks[i].className = tablinks[i].className.replace(" active", ""); }}
-            document.getElementById(tabName).style.display = "block";
-            evt.currentTarget.className += " active";
+            let i, tabcontent = document.getElementsByClassName("tab-content"), tablinks = document.getElementsByClassName("tab-button");
+            for (i = 0; i < tabcontent.length; i++) tabcontent[i].style.display = "none";
+            for (i = 0; i < tablinks.length; i++) tablinks[i].className = tablinks[i].className.replace(" active", "");
+            document.getElementById(tabName).style.display = "block"; evt.currentTarget.className += " active";
         }}
-        
-        const mainImage = document.getElementById('main-image');
-        const mainFilename = document.getElementById('main-filename');
-        const thumbnails = document.querySelectorAll('.thumbnail');
+        const mainImage = document.getElementById('main-image'), mainFilename = document.getElementById('main-filename'), thumbnails = document.querySelectorAll('.thumbnail');
         if (thumbnails.length > 0) thumbnails[0].classList.add('active');
-
-        function selectThumbnail(selectedThumb, imgSrc, filename) {{
-            mainImage.src = imgSrc; mainFilename.textContent = filename;
-            thumbnails.forEach(t => t.classList.remove('active'));
-            selectedThumb.classList.add('active');
+        function selectThumbnail(thumb, src, name) {{
+            mainImage.src = src; mainFilename.textContent = name;
+            thumbnails.forEach(t => t.classList.remove('active')); thumb.classList.add('active');
         }};
     </script>
     """
     return html
 
-def generate_outputs_api(omni_files, omni_prompt, output_formats, layouts, fonts, num_columns, page_w_mult, page_h_mult, progress=gr.Progress(track_tqdm=True)):
-    """🚀 The main entry point that orchestrates the entire multi-modal generation process."""
-    if not omni_prompt and not omni_files: raise gr.Error("Please provide a prompt or upload at least one file.")
+def generate_outputs_api(omni_file, omni_prompt, output_formats, layouts, fonts, num_columns, page_w_mult, page_h_mult, progress=gr.Progress(track_tqdm=True)):
+    if not omni_prompt and not omni_file: raise gr.Error("Please provide a prompt or upload at least one file.")
     if not output_formats: raise gr.Error("Please select at least one output format.")
     
     shutil.rmtree(OUTPUT_DIR, ignore_errors=True); shutil.rmtree(PREVIEW_DIR, ignore_errors=True)
     OUTPUT_DIR.mkdir(); PREVIEW_DIR.mkdir()
 
     md_content = ""
-    if omni_files:
-        temp_paths = []
-        for f in omni_files:
-            temp_path = UPLOAD_DIR / Path(f.name).name
-            shutil.copyfile(f.name, temp_path)
-            temp_paths.append(temp_path)
-        
-        file_path = temp_paths[0]
-        file_ext = file_path.suffix.lower()
-
-        if file_ext == '.md': md_content = "\n\n".join([p.read_text(encoding='utf-8') for p in temp_paths if p.suffix.lower() == '.md'])
-        elif file_ext == '.pdf': md_content = process_pdf_input(file_path, omni_prompt or "Extract text", progress)
-        elif file_ext in ['.png', '.jpg', '.jpeg']: md_content = process_image_input(file_path, omni_prompt or "Describe image")
-        elif file_ext in ['.wav', '.mp3']: md_content = process_audio_input(file_path, omni_prompt or "Summarize transcription")
+    if omni_file:
+        temp_path = UPLOAD_DIR / Path(omni_file.name).name
+        shutil.copyfile(omni_file.name, temp_path)
+        file_ext = temp_path.suffix.lower()
+
+        if file_ext == '.md': md_content = temp_path.read_text(encoding='utf-8')
+        elif file_ext == '.pdf': md_content = process_pdf_input(temp_path, omni_prompt or "Extract text", progress)
+        elif file_ext in ['.png', '.jpg', '.jpeg']: md_content = process_image_input(temp_path, omni_prompt or "Describe image")
+        elif file_ext in ['.wav', '.mp3']: md_content = process_audio_input(temp_path, omni_prompt or "Summarize transcription")
     elif omni_prompt:
         md_content = process_text_input(omni_prompt)
     
@@ -319,8 +295,7 @@ def generate_outputs_api(omni_files, omni_prompt, output_formats, layouts, fonts
                     pagesize = LAYOUTS[layout_name]["size"]
                     final_pagesize = (pagesize[0] * page_w_mult, pagesize[1] * page_h_mult)
                     pdf_buffer = create_pdf(md_content, font_name, EMOJI_FONT_NAME, final_pagesize, num_columns)
-                    filename = f"Document_{time_str}.pdf"
-                    output_path = OUTPUT_DIR / filename
+                    filename = f"Document_{time_str}.pdf"; output_path = OUTPUT_DIR / filename
                     with open(output_path, "wb") as f: f.write(pdf_buffer.getvalue())
                     generated_files.append(output_path); pdf_files_for_gallery.append(output_path)
         elif format_choice == "DOCX":
@@ -336,39 +311,17 @@ def generate_outputs_api(omni_files, omni_prompt, output_formats, layouts, fonts
 
 # --- 🎨 Gradio UI Definition ---
 AVAILABLE_FONTS, EMOJI_FONT_NAME = register_local_fonts()
-SAMPLE_MARKDOWN = """# Deities Guide: Mythology and Moral Lessons 
-
-1. 📜 **Introduction**
-   - **Purpose**: Explore deities, spirits, saints, and beings with their epic stories and morals!
-   - **Usage**: A guide for learning and storytelling across traditions. ️
-   - **Themes**: Justice ⚖️, faith 🙏, hubris 🏛️, redemption ✨, cosmic order 🌌.
-
-# ⚔️ Arthurian Legends
- - **Merlin, Morgan le Fay, Arthur**: Mentor 🧙, rival 🧙‍♀️, son 👑.
- - **Relation**: Family tests loyalty 🤝.
- - **Lesson**: Honor 🎖️ vs. betrayal 🗡️.
-"""
 
 with gr.Blocks(theme=gr.themes.Soft(), title="Omni-Model Document Generator") as demo:
     gr.Markdown("# 🧠 Omni-Model Document Generator")
-    gr.Markdown("Provide a prompt, or upload a file (MD, PDF, Image, Audio). A virtual AI will process it, and you can generate documents from the result.")
+    gr.Markdown("Provide a prompt or upload a file (MD, PDF, Image, Audio). A virtual AI will process it, and you can generate documents from the result.")
 
     with gr.Row():
         with gr.Column(scale=1):
-            with gr.Tabs():
-                with gr.TabItem("💬 Text"):
-                    text_prompt = gr.Textbox(label="Prompt", lines=5, placeholder="Ask a question or provide instructions...")
-                with gr.TabItem("🖼️ Image"):
-                    image_prompt = gr.Textbox(label="Image Prompt", lines=2, placeholder="e.g., Describe this picture")
-                    image_file = gr.File(label="Upload Image", file_types=["image"])
-                with gr.TabItem("🎤 Audio"):
-                    audio_prompt = gr.Textbox(label="Audio Prompt", lines=2, placeholder="e.g., Summarize this audio")
-                    audio_file = gr.File(label="Upload Audio", file_types=[".wav", ".mp3"])
-                with gr.TabItem("📄 Document"):
-                    doc_prompt = gr.Textbox(label="Document Prompt", lines=2, placeholder="e.g., Extract text from this PDF")
-                    doc_file = gr.File(label="Upload MD or PDF", file_types=[".md", ".pdf"])
+            gr.Markdown("### ⚙️ Input & Settings")
+            omni_prompt = gr.Textbox(label="Prompt", lines=3, placeholder="Ask a question or provide instructions...")
+            omni_file = gr.File(label="Upload File (Optional)", file_types=["image", ".wav", ".mp3", ".md", ".pdf"])
             
-            gr.Markdown("### 📄 Output Settings")
             output_formats = gr.CheckboxGroup(choices=["PDF", "DOCX", "XLSX"], label="Select Output Formats", value=["PDF"])
             
             with gr.Accordion("PDF Customization", open=False):
@@ -386,17 +339,8 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Omni-Model Document Generator") as
             gr.Markdown("### 🗂️ Generated Files")
             file_explorer_output = gr.HTML(label="File Explorer & Gallery")
 
-    def master_process(p1, p2, p3, p4, f1, f2, f3, f4, *args):
-        # Determine active tab and route to the API
-        if f1: return generate_outputs_api([f1], p1 or "Describe this text", *args)
-        if f2: return generate_outputs_api([f2], p2 or "Describe this image", *args)
-        if f3: return generate_outputs_api([f3], p3 or "Summarize this audio", *args)
-        if f4: return generate_outputs_api([f4], p4 or "Process this document", *args)
-        if p1: return generate_outputs_api(None, p1, *args)
-        raise gr.Error("Please provide an input in one of the tabs.")
-
-    generate_btn.click(fn=master_process, 
-                       inputs=[text_prompt, image_prompt, audio_prompt, doc_prompt, text_prompt, image_file, audio_file, doc_file, output_formats, selected_layouts, selected_fonts, num_columns_slider, page_w_mult_slider, page_h_mult_slider], 
+    generate_btn.click(fn=generate_outputs_api, 
+                       inputs=[omni_file, omni_prompt, output_formats, selected_layouts, selected_fonts, num_columns_slider, page_w_mult_slider, page_h_mult_slider], 
                        outputs=[ai_response_output, file_explorer_output])
 
 if __name__ == "__main__":