[update] app.py

app.py

@@ -460,7 +460,7 @@ def start_generation(pdf_file, model_name_t, model_name_v, template_root,
                     qwen_api_key, zhipuai_api_key, openrouter_api_key):
     """Start generation process"""
     if pdf_file is None:
-        return "❌ Please upload a PDF file", gr.update(visible=False), "", "", gr.update(), gr.update()
+        return "❌ Please upload a PDF file", gr.update(visible=False), "", "", gr.update(), gr.update(), ""
     
     # Validate API keys
     validation_errors = validate_api_keys(
@@ -470,7 +470,7 @@ def start_generation(pdf_file, model_name_t, model_name_v, template_root,
     
     if validation_errors:
         error_msg = "❌ API Key Validation Failed:\n" + "\n".join(f"• {error}" for error in validation_errors)
-        return error_msg, gr.update(visible=False), "", "", gr.update(), gr.update()
+        return error_msg, gr.update(visible=False), "", "", gr.update(), gr.update(), ""
     
     state.reset()
     
@@ -489,7 +489,7 @@ def start_generation(pdf_file, model_name_t, model_name_v, template_root,
             try:
                 matched = matching(template_requirement)
             except Exception as e:
-                return f"❌ Template recommendation failed: {e}", gr.update(visible=False), "", "", gr.update(choices=[], value=None), gr.update(visible=False, value="")
+                return f"❌ Template recommendation failed: {e}", gr.update(visible=False), "", "", gr.update(choices=[], value=None), gr.update(visible=False, value=""), ""
             
             html_finder_ = HtmlFinder()
             with open('templates/template_link.json','r') as f:
@@ -510,14 +510,14 @@ def start_generation(pdf_file, model_name_t, model_name_v, template_root,
                     continue
             
             if not previews:
-                return "❌ No previewable templates found", gr.update(visible=False), "", "", gr.update(choices=[], value=None), gr.update(visible=False, value="")
+                return "❌ No previewable templates found", gr.update(visible=False), "", "", gr.update(choices=[], value=None), gr.update(visible=False, value=""), ""
             
             md_lines = ["### 🔍 Please select a template to preview before clicking **Start Generation**", ""]
             for name, _, url in previews:
                 md_lines.append(f"- **{name}** → [{url}]({url})")
             md = "\n".join(md_lines)
             
-            return "Recommended 3 templates, please select one to continue", gr.update(visible=False), "", "", gr.update(choices=[n for n, _, _ in previews], value=None), gr.update(visible=True, value=md)
+            return "Recommended 3 templates, please select one to continue", gr.update(visible=False), "", "", gr.update(choices=[n for n, _, _ in previews], value=None), gr.update(visible=True, value=md), ""
         
         template_dir = os.path.join(template_root, template_choice_value)
     
@@ -545,7 +545,7 @@ def start_generation(pdf_file, model_name_t, model_name_v, template_root,
     )
     
     if not args.template_dir:
-        return "❌ Please select a template", gr.update(visible=False), "", "", gr.update(), gr.update()
+        return "❌ Please select a template", gr.update(visible=False), "", "", gr.update(), gr.update(), ""
     
     if not args.template_file:
         html_finder_ = HtmlFinder()
@@ -627,14 +627,23 @@ def start_generation(pdf_file, model_name_t, model_name_v, template_root,
             section_display_md,       # Markdown format
             section_display_json,     # JSON format (hidden)
             gr.update(),
-            gr.update(visible=False, value="")
+            gr.update(visible=False, value=""),
+            ""
         )
         
     except Exception as e:
         import traceback
         error_msg = f"❌ Generation failed: {str(e)}\n{traceback.format_exc()}"
-        return error_msg, gr.update(visible=False), "", "", gr.update(), gr.update()
+        return error_msg, gr.update(visible=False), "", "", gr.update(), gr.update(), ""
 
+def create_dynamic_page_dict(sections):
+        poster_dict = {
+            "title": "Title of the paper",
+            "authors": "Authors of the paper",
+            "affiliation": "Affiliation of the authors",
+        }
+        poster_dict.update(sections)
+        return poster_dict
 def generate_section_initial():
     """Generate initial Section"""
     import yaml
@@ -659,16 +668,6 @@ def generate_section_initial():
     response = state.planner.planner_agent.step(prompt)
     input_token, output_token = account_token(response)
     generated_section = get_json_from_response(response.msgs[0].content)
-    
-    def create_dynamic_page_dict(sections):
-        poster_dict = {
-            "title": "Title of the paper",
-            "authors": "Authors of the paper",
-            "affiliation": "Affiliation of the authors",
-        }
-        poster_dict.update(sections)
-        return poster_dict
-    
     generated_section = create_dynamic_page_dict(generated_section)
     state.generated_section = generated_section
     
@@ -691,6 +690,7 @@ def submit_section_feedback(feedback_text):
             "",  # section_feedback_input clear
             gr.update(visible=False),  # feedback_section hide
             fc_section_visible,  # feedback_full_content show
+            fc_display_visible,  # full_content_display_md show
             fc_display_md,  # full_content_display_md content
             fc_display_json,  # full_content_display_json content
             fc_feedback_visible  # full_content_feedback_input show
@@ -711,6 +711,7 @@ def submit_section_feedback(feedback_text):
     state.total_output_tokens_t += output_token
     
     generated_section = get_json_from_response(response.msgs[0].content)
+    generated_section = create_dynamic_page_dict(generated_section)
     state.generated_section = generated_section
     
     generated_path = f'project_contents/{state.args.paper_name}_generated_section.json'
@@ -728,6 +729,7 @@ def submit_section_feedback(feedback_text):
         "",  # Clear input box
         gr.update(visible=True),  # feedback_section keep visible
         gr.update(visible=False),  # feedback_full_content keep hidden
+        gr.update(visible=False),  # full_content_display_md keep hidden
         "",  # full_content_display_md content
         "",  # full_content_display_json content
         gr.update(visible=False)  # full_content_feedback_input keep hidden
@@ -1233,6 +1235,8 @@ with gr.Blocks(title="Paper2ProjectPage Generator", theme=gr.themes.Soft(), css=
             gr.Markdown("""
             **⚠️ Security Notice**: Your API keys are only stored in memory during the session and are never saved to disk.
             
+            **💡 Note**: You only need to fill in ONE API key corresponding to the models you selected below.
+            
             **📋 How to get API keys:**
             - **OpenAI**: Get your API key from [OpenAI Platform](https://platform.openai.com/api-keys)
             - **Gemini**: Get your API key from [Google AI Studio](https://aistudio.google.com/app/apikey)
@@ -1337,59 +1341,46 @@ with gr.Blocks(title="Paper2ProjectPage Generator", theme=gr.themes.Soft(), css=
                     info="Select model for vision processing"
                 )
             
-            gr.Markdown("### 📁 Path Configuration")
+            # Hidden backend configuration fields
             template_root = gr.Textbox(
                 label="Template Root",
                 value="templates",
-                info="Root directory for templates"
+                visible=False
             )
             template_dir = gr.Textbox(
                 label="Template Directory",
                 value="",
-                info="Selected template directory (optional)"
+                visible=False
             )
             template_file = gr.Textbox(
                 label="Template File",
                 value="",
-                info="Specific template file path (optional)"
-            )
-            template_choice = gr.Radio(
-                label="Recommended Templates",
-                choices=[],
-                value=None,
-                info="Select from recommended templates",
-                visible=True
+                visible=False
             )
             output_dir = gr.Textbox(
                 label="Output Directory",
                 value="generated_project_pages",
-                info="Directory for output files"
+                visible=False
             )
             style_preference = gr.Textbox(
                 label="Style Preference JSON",
                 value="",
-                info="Style preference JSON file path (optional)"
+                visible=False
             )
             tmp_dir = gr.Textbox(
                 label="Temporary Directory",
                 value="tmp",
-                info="Directory for temporary files"
-            )
-
-            template_preview_links = gr.Markdown(
-                label="Template Preview Links",
-                value="",
                 visible=False
             )
             
-            # ===== Hidden parameters with default values =====
+            # Hidden parameters with default values
             resume = gr.Radio(
                 label="Resume From Step",
                 choices=['parse_pdf', 'generate_content','full_content_check', 'generate_html', 'html_check','modify_table','html_feedback'],
                 value='parse_pdf',
                 visible=False
             )
-
+            
             human_input = gr.Radio(
                 label="Enable Human Feedback",
                 choices=[0, 1],
@@ -1397,65 +1388,100 @@ with gr.Blocks(title="Paper2ProjectPage Generator", theme=gr.themes.Soft(), css=
                 visible=False
             )
             
-        with gr.Column(scale=1):
-            gr.Markdown("### 🎨 Style Configuration")
-            
-            background_color = gr.Radio(
-                label="Background Color",
-                choices=["light", "dark"],
-                value="light",
-                info="Background color theme"
+            full_content_check_times = gr.Number(
+                label="Full Content Check Times",
+                value=1,
+                precision=0,
+                visible=False
             )
             
-            has_navigation = gr.Radio(
-                label="Has Navigation",
-                choices=["yes", "no"],
-                value="yes",
-                info="Include navigation bar"
+            html_check_times = gr.Number(
+                label="HTML Check Times",
+                value=1,
+                precision=0,
+                visible=False
             )
             
-            has_hero_section = gr.Radio(
-                label="Has Hero Section",
-                choices=["yes", "no"],
-                value="yes",
-                info="Include hero/header section"
-            )
+        with gr.Column(scale=1):
+            gr.Markdown("""
+            ### 💡 User Guide
             
-            title_color = gr.Radio(
-                label="Title Color",
-                choices=["pure", "colorful"],
-                value="pure",
-                info="Title color style"
-            )
+            1. **Upload PDF**: Select your research paper PDF file
+            2. **Configure API Key**: Fill in ONE API key for your selected models
+            3. **Select Models**: Choose text and vision models from dropdowns
+            4. **Configure Style**: Adjust style preferences below
+            5. **Start Generation**: Click the "Start Generation" button
+            6. **Three-Stage Feedback**:
+               - 📝 **Section Feedback**: Review the generated page structure (Markdown preview + JSON data)
+               - 📄 **Full Content Feedback**: Review the generated complete content (Markdown preview + JSON data)
+               - 🌐 **HTML Feedback**: View the generated webpage in a new tab
+            7. **Download Results**: Download the complete project archive after completion
             
-            page_density = gr.Radio(
-                label="Page Density",
-                choices=["spacious", "compact"],
-                value="spacious",
-                info="Page spacing density"
-            )
+            ⚠️ **Tips**: 
+            - Each stage supports multiple rounds of feedback until you're satisfied
+            - Enter 'yes' to proceed to the next stage
+            - The final ZIP download includes the complete project folder with all resources
+            """)
             
-            image_layout = gr.Radio(
-                label="Image Layout",
-                choices=["rotation", "parallelism"],
-                value="parallelism",
-                info="Image layout style"
-            )
+            gr.Markdown("### 🎨 Style Configuration")
             
-            gr.Markdown("### ⚙️ Advanced Options")
+            with gr.Row():
+                with gr.Column():
+                    background_color = gr.Radio(
+                        label="Background Color",
+                        choices=["light", "dark"],
+                        value="light",
+                        info="Background color theme"
+                    )
+                    
+                    has_navigation = gr.Radio(
+                        label="Has Navigation",
+                        choices=["yes", "no"],
+                        value="yes",
+                        info="Include navigation bar"
+                    )
+                    
+                    has_hero_section = gr.Radio(
+                        label="Has Hero Section",
+                        choices=["yes", "no"],
+                        value="yes",
+                        info="Include hero/header section"
+                    )
+                
+                with gr.Column():
+                    title_color = gr.Radio(
+                        label="Title Color",
+                        choices=["pure", "colorful"],
+                        value="pure",
+                        info="Title color style"
+                    )
+                    
+                    page_density = gr.Radio(
+                        label="Page Density",
+                        choices=["spacious", "compact"],
+                        value="spacious",
+                        info="Page spacing density"
+                    )
+                    
+                    image_layout = gr.Radio(
+                        label="Image Layout",
+                        choices=["rotation", "parallelism"],
+                        value="parallelism",
+                        info="Image layout style"
+                    )
             
-            full_content_check_times = gr.Number(
-                label="Full Content Check Times",
-                value=1,
-                precision=0,
-                info="Number of full content validation checks"
+            gr.Markdown("### 📁 Template Selection")
+            
+            template_choice = gr.Radio(
+                label="Recommended Templates",
+                choices=[],
+                value=None,
+                info="Select from recommended templates"
             )
             
-            html_check_times = gr.Number(
-                label="HTML Check Times",
-                value=1,
-                precision=0,
-                info="Number of HTML validation checks"
+            template_preview_links = gr.Markdown(
+                value="",
+                visible=False
             )
     
     # Start Generation Button
@@ -1557,28 +1583,6 @@ with gr.Blocks(title="Paper2ProjectPage Generator", theme=gr.themes.Soft(), css=
         interactive=False
     )
     
-    gr.Markdown("""
-    ---
-    ### 💡 User Guide
-    
-    1. **Upload PDF**: Select your research paper PDF file
-    2. **Configure Parameters**: Adjust model, path, and style settings as needed
-    3. **Start Generation**: Click the "Start Generation" button
-    4. **Three-Stage Feedback**:
-       - 📝 **Section Feedback**: Review the generated page structure (Markdown preview + JSON data), provide feedback or enter 'yes' to continue
-       - 📄 **Full Content Feedback**: Review the generated complete content (Markdown preview + JSON data), provide feedback or enter 'yes' to continue
-       - 🌐 **HTML Feedback**: View the generated webpage in a new tab, provide feedback or enter 'yes' to finalize
-    5. **Download Results**: Download the complete project archive after completion
-    
-    ⚠️ **Tips**: 
-    - Each stage supports multiple rounds of feedback until you're satisfied
-    - Section and Full Content stages offer **Markdown preview** and **JSON raw data** viewing options
-    - Markdown preview is more visually appealing, JSON data shows complete structure
-    - HTML stage requires clicking "Open Preview in New Tab" to view the full page in browser
-    - Enter 'yes' to indicate satisfaction and proceed to the next stage
-    - The final ZIP download includes the complete project folder with all resources
-    """)
-    
     # Bind Events
     start_btn.click(
         fn=start_generation,
@@ -1598,7 +1602,7 @@ with gr.Blocks(title="Paper2ProjectPage Generator", theme=gr.themes.Soft(), css=
             section_display_json,
             template_choice,
             template_preview_links,
-            # section_feedback_input
+            section_feedback_input
         ]
     )
     
@@ -1613,6 +1617,7 @@ with gr.Blocks(title="Paper2ProjectPage Generator", theme=gr.themes.Soft(), css=
             feedback_section,
             feedback_full_content,
             full_content_display_md,
+            full_content_display_md,
             full_content_display_json,
             full_content_feedback_input
         ]
@@ -1646,13 +1651,10 @@ with gr.Blocks(title="Paper2ProjectPage Generator", theme=gr.themes.Soft(), css=
             html_file_output
         ]
     )
-    def dummy_open_preview(url):
-        """Dummy function to pass URL to JavaScript"""
-        return None
     
     # Open Preview Button - Use JavaScript to open in new tab
     open_preview_btn.click(
-        fn=dummy_open_preview,
+        fn=None,
         inputs=[preview_url_state],
         outputs=None,
         js="(url) => window.open(url, '_blank')"
@@ -1665,4 +1667,4 @@ if __name__ == "__main__":
         server_port=7860,
         share=False,
         show_error=True
-    )
+    )