-
- {'์ด ํ
๋ง' if language == 'Korean' else 'Total Themes'}
- {total_themes}
-
-
- {'์ด ์กฐํ์' if language == 'Korean' else 'Total Views'}
- {total_views}
-
-
- {'์ด ์ฌ์ฉ์' if language == 'Korean' else 'Total Uses'}
- {total_uses}
-
-
'
- for theme in themes:
- cards_html += format_theme_card(theme, language)
- cards_html += '
'
-
- # JavaScript for interactions
- js_script = """
-"""
-
- return stats_html + cards_html + js_script
-
-def load_css():
- """Load CSS from external file"""
- try:
- with open('css.css', 'r', encoding='utf-8') as f:
- return f.read()
- except FileNotFoundError:
- logger.error("css.css file not found")
- return ""
-
+ """Format stages display for screenplay"""
+ markdown = "## ๐ฌ Production Progress\n\n"
+
+ # Progress summary
+ completed = sum(1 for s in stages if s.get('status') == 'complete')
+ total = len(stages)
+ markdown += f"**Progress: {completed}/{total} stages complete**\n\n"
+
+ # Page count if available
+ total_pages = sum(s.get('page_count', 0) for s in stages if s.get('page_count'))
+ if total_pages > 0:
+ markdown += f"**Current Page Count: {total_pages:.1f} pages**\n\n"
+
+ markdown += "---\n\n"
+
+ # Stage details
+ current_act = None
+ for i, stage in enumerate(stages):
+ status_icon = "โ
" if stage['status'] == 'complete' else "๐" if stage['status'] == 'active' else "โณ"
+
+ # Group by acts
+ if 'Act' in stage.get('name', ''):
+ act_match = re.search(r'Act (\w+)', stage['name'])
+ if act_match and act_match.group(1) != current_act:
+ current_act = act_match.group(1)
+ markdown += f"\n### ๐ Act {current_act}\n\n"
+
+ markdown += f"{status_icon} **{stage['name']}**"
+
+ if stage.get('page_count', 0) > 0:
+ markdown += f" ({stage['page_count']:.1f} pages)"
+
+ markdown += "\n"
+
+ if stage['content'] and stage['status'] == 'complete':
+ preview_length = 200
+ preview = stage['content'][:preview_length] + "..." if len(stage['content']) > preview_length else stage['content']
+ markdown += f"> {preview}\n\n"
+ elif stage['status'] == 'active':
+ markdown += "> *In progress...*\n\n"
+
+ return markdown
+
+def process_query(query: str, screenplay_type: str, genre: str, language: str,
+ session_id: Optional[str] = None) -> Generator[Tuple[str, str, str, str], None, None]:
+ """Main query processing function"""
+ if not query.strip():
+ yield "", "", "โ Please enter a screenplay concept.", session_id
+ return
+
+ system = ScreenplayGenerationSystem()
+ stages_markdown = ""
+ screenplay_display = ""
+
+ for status, stages, current_session_id in system.process_screenplay_stream(
+ query, screenplay_type, genre, language, session_id
+ ):
+ stages_markdown = format_stages_display(stages)
+
+ # Get screenplay content when available
+ if stages and all(s.get("status") == "complete" for s in stages[-4:]):
+ screenplay_text = ScreenplayDatabase.get_screenplay_content(current_session_id)
+ screenplay_display = format_screenplay_display(screenplay_text)
+
+ yield stages_markdown, screenplay_display, status or "๐ Processing...", current_session_id
+
+def get_active_sessions() -> List[str]:
+ """Get active screenplay sessions"""
+ sessions = ScreenplayDatabase.get_active_sessions()
+ return [
+ f"{s['session_id'][:8]}... - {s.get('title', s['user_query'][:30])}... "
+ f"({s['screenplay_type']}/{s['genre']}) [{s['total_pages']:.1f} pages]"
+ for s in sessions
+ ]
+
+def export_screenplay_pdf(screenplay_text: str, title: str, session_id: str) -> str:
+ """Export screenplay to PDF format"""
+ # This would use a library like reportlab to create industry-standard PDF
+ # For now, returning a placeholder
+ pdf_path = f"screenplay_{session_id[:8]}.pdf"
+ # PDF generation logic would go here
+ return pdf_path
+
+def export_screenplay_fdx(screenplay_text: str, title: str, session_id: str) -> str:
+ """Export to Final Draft format"""
+ # This would create .fdx XML format
+ fdx_path = f"screenplay_{session_id[:8]}.fdx"
+ # FDX generation logic would go here
+ return fdx_path
+
+def download_screenplay(screenplay_text: str, format_type: str, title: str,
+ session_id: str) -> Optional[str]:
+ """Generate screenplay download file"""
+ if not screenplay_text or not session_id:
+ return None
+
+ timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+
+ try:
+ if format_type == "PDF":
+ return export_screenplay_pdf(screenplay_text, title, session_id)
+ elif format_type == "FDX":
+ return export_screenplay_fdx(screenplay_text, title, session_id)
+ elif format_type == "FOUNTAIN":
+ filepath = f"screenplay_{session_id[:8]}_{timestamp}.fountain"
+ with open(filepath, 'w', encoding='utf-8') as f:
+ f.write(screenplay_text)
+ return filepath
+ else: # TXT
+ filepath = f"screenplay_{session_id[:8]}_{timestamp}.txt"
+ with open(filepath, 'w', encoding='utf-8') as f:
+ f.write(f"Title: {title}\n")
+ f.write("=" * 50 + "\n\n")
+ f.write(screenplay_text)
+ return filepath
+ except Exception as e:
+ logger.error(f"Download generation failed: {e}")
+ return None
+
# Create Gradio interface
def create_interface():
- combined_css = load_css()
- # Using Soft theme with safe color options
- with gr.Blocks(theme=gr.themes.Soft(), css=combined_css, title="AGI NOVEL Generator") as interface:
- gr.HTML("""
-
-
-
-
-
-
-
-
- ๐ฒ Novel Theme Random Generator: This system can generate up to approximately 170 quadrillion (1.7 ร 10ยนโท) unique novel themes.
- Even writing 100 novels per day, it would take 4.6 million years to exhaust all combinations.
- Click the "Random" button to explore infinite creative possibilities!
-
-
-
- โฑ๏ธ Note: Creating a complete novel takes approximately 20 minutes. If your web session disconnects, you can restore your work using the "Session Recovery" feature.
-
-
-
- ๐ฏ Core Innovation: Not fragmented texts from multiple writers,
- but a genuine full-length novel written consistently by a single author from beginning to end.
-
-
Library feature coming soon...
"
+ )
+
+ # Event handlers
+ def handle_submit(query, s_type, genre, lang, session_id):
+ if not query:
+ yield "", "", "โ Please enter a concept", session_id
+ return
- # Random Theme Library Tab
- with gr.Tab("๐ฒ Random Theme Library", elem_id="library_tab"):
- with gr.Column():
- gr.Markdown("""
- ### ๐ Random Theme Library
-
- Browse through all randomly generated themes. Each theme is unique and can be used to create a novel.
- """)
-
- with gr.Row():
- library_search = gr.Textbox(
- label="Search Themes",
- placeholder="Search by keywords...",
- elem_classes=["library-search"],
- scale=2
- )
- library_language_filter = gr.Radio(
- choices=["All", "English", "Korean"],
- value="All",
- label="Filter by Language",
- scale=2
- )
- library_refresh_btn = gr.Button("๐ Refresh", scale=1)
-
- library_display = gr.HTML(
- value=get_theme_library_display(),
- elem_id="library-display"
- )
-
- # Hidden components for theme interaction
- selected_theme_for_action = gr.Textbox(visible=False, elem_id="selected_theme_for_action")
- action_type = gr.Textbox(visible=False, elem_id="action_type")
- trigger_action = gr.Button("Trigger Action", visible=False, elem_id="trigger_action")
-
- # Event handlers
- def refresh_sessions():
- try:
- sessions = get_active_sessions("English")
- return gr.update(choices=sessions)
- except Exception as e:
- logger.error(f"Session refresh error: {str(e)}")
- return gr.update(choices=[])
-
- def handle_auto_recover(language):
- session_id, message = auto_recover_session(language)
- return session_id, message
-
- def handle_random_theme(language):
- """Handle random theme generation with library storage"""
- theme = generate_random_theme(language)
- return theme
-
- def refresh_library(language_filter, search_query):
- """Refresh theme library display"""
- lang = None if language_filter == "All" else language_filter
- return get_theme_library_display(lang, search_query)
-
- def handle_library_action(theme_id, action):
- """Handle theme library actions"""
- if not theme_id:
- return gr.update(), gr.update()
-
- if action == "use":
- # Handle use action
- theme_data = NovelDatabase.get_theme_by_id(theme_id)
- if theme_data:
- NovelDatabase.update_theme_used_count(theme_id)
- NovelDatabase.update_theme_view_count(theme_id) # Also update view count
- return (
- gr.update(value=theme_data.get('theme_text', '')), # query_input
- f"Theme #{theme_id[:8]} loaded" # status_text
- )
-
- return gr.update(), gr.update()
-
- # Event connections
- submit_btn.click(
- fn=process_query,
- inputs=[query_input, language_select, current_session_id],
- outputs=[stages_display, novel_output, status_text, current_session_id, novel_text_state] # novel_text_state ์ถ๊ฐ
- )
-
- resume_btn.click(
- fn=lambda x: x.split("...")[0] if x and "..." in x else x,
- inputs=[session_dropdown],
- outputs=[current_session_id]
- ).then(
- fn=resume_session,
- inputs=[current_session_id, language_select],
- outputs=[stages_display, novel_output, status_text, current_session_id, novel_text_state] # novel_text_state ์ถ๊ฐ
- )
-
- auto_recover_btn.click(
- fn=handle_auto_recover,
- inputs=[language_select],
- outputs=[current_session_id, status_text]
- ).then(
- fn=resume_session,
- inputs=[current_session_id, language_select],
- outputs=[stages_display, novel_output, status_text, current_session_id, novel_text_state] # novel_text_state ์ถ๊ฐ
- )
-
-
- refresh_btn.click(
- fn=refresh_sessions,
- outputs=[session_dropdown]
- )
-
- clear_btn.click(
- fn=lambda: ("", "", "โจ Ready to begin your literary journey", "", None, ""), # ๋น ๋ฌธ์์ด ์ถ๊ฐ
- outputs=[stages_display, novel_output, status_text, novel_text_state, current_session_id, novel_text_state]
- )
-
- random_btn.click(
- fn=handle_random_theme,
- inputs=[language_select],
- outputs=[query_input],
- queue=False
- )
-
- # Library event handlers
- library_refresh_btn.click(
- fn=refresh_library,
- inputs=[library_language_filter, library_search],
- outputs=[library_display]
- )
-
- library_search.change(
- fn=refresh_library,
- inputs=[library_language_filter, library_search],
- outputs=[library_display]
- )
-
- library_language_filter.change(
- fn=refresh_library,
- inputs=[library_language_filter, library_search],
- outputs=[library_display]
- )
-
- # Handle clicks on library display - using trigger button
- trigger_action.click(
- fn=handle_library_action,
- inputs=[selected_theme_for_action, action_type],
- outputs=[query_input, status_text]
- )
-
-
- download_btn.click(
- fn=handle_download,
- inputs=[format_select, language_select, current_session_id, novel_text_state],
- outputs=[download_file]
- )
-
- # Load sessions and library on start
- def initialize_interface():
- # Sync with HF dataset on startup
- if 'hf_manager' in globals() and hf_manager.token:
- hf_manager.sync_with_local_db()
-
- return refresh_sessions(), refresh_library("All", "")
-
- interface.load(
- fn=initialize_interface,
- outputs=[session_dropdown, library_display]
- )
+ yield from process_query(query, s_type, genre, lang, session_id)
+
+ def handle_random(s_type, genre, lang):
+ return generate_random_screenplay_theme(s_type, genre, lang)
+
+ def handle_download(screenplay_text, format_type, session_id):
+ if not screenplay_text or not session_id:
+ return gr.update(visible=False)
+
+ # Get title from database
+ session = ScreenplayDatabase.get_session(session_id)
+ title = session.get('title', 'Untitled') if session else 'Untitled'
+
+ file_path = download_screenplay(screenplay_text, format_type, title, session_id)
+ if file_path and os.path.exists(file_path):
+ return gr.update(value=file_path, visible=True)
+ return gr.update(visible=False)
+
+ # Connect events
+ submit_btn.click(
+ fn=handle_submit,
+ inputs=[query_input, screenplay_type, genre_select, language_select, current_session_id],
+ outputs=[stages_display, screenplay_output, status_text, current_session_id]
+ )
+
+ random_btn.click(
+ fn=handle_random,
+ inputs=[screenplay_type, genre_select, language_select],
+ outputs=[query_input]
+ )
+
+ clear_btn.click(
+ fn=lambda: ("", "", "Ready to create your screenplay", None),
+ outputs=[stages_display, screenplay_output, status_text, current_session_id]
+ )
+
+ refresh_btn.click(
+ fn=get_active_sessions,
+ outputs=[session_dropdown]
+ )
+
+ download_btn.click(
+ fn=handle_download,
+ inputs=[screenplay_output, format_select, current_session_id],
+ outputs=[download_file]
+ )
+
+ # Load sessions on start
+ interface.load(
+ fn=get_active_sessions,
+ outputs=[session_dropdown]
+ )
- return interface
-# Initialize HF Dataset Manager as global variable
-hf_manager = None
+ return interface
# Main function
if __name__ == "__main__":
- logger.info("AGI NOVEL Generator v2.0 Starting...")
- logger.info("=" * 60)
-
- # Environment check
- logger.info(f"API Endpoint: {API_URL}")
- logger.info(f"Target Length: {TARGET_WORDS:,} words")
- logger.info(f"Minimum Words per Part: {MIN_WORDS_PER_PART:,} words")
- logger.info("System Features: Single writer + Immediate part-by-part critique")
-
- if BRAVE_SEARCH_API_KEY:
- logger.info("Web search enabled.")
- else:
- logger.warning("Web search disabled.")
-
- if DOCX_AVAILABLE:
- logger.info("DOCX export enabled.")
- else:
- logger.warning("DOCX export disabled.")
-
- logger.info("=" * 60)
-
- # Initialize database
- logger.info("Initializing database...")
- NovelDatabase.init_db()
- logger.info("Database initialization complete.")
-
- # Initialize HF Dataset Manager
- logger.info("Initializing HuggingFace dataset manager...")
- hf_manager = HFDatasetManager()
-
- if hf_manager.token:
- logger.info("HuggingFace authentication successful.")
- # Sync with HF dataset on startup
- hf_manager.sync_with_local_db()
- else:
- logger.warning("HuggingFace token not found. Theme persistence will be local only.")
-
- # Create and launch interface
- interface = create_interface()
-
- interface.launch(
- server_name="0.0.0.0",
- server_port=7860,
- share=False,
- debug=True
- )
\ No newline at end of file
+ logger.info("Screenplay Generator Starting...")
+ logger.info("=" * 60)
+
+ # Environment check
+ logger.info(f"API Endpoint: {API_URL}")
+ logger.info("Screenplay Types Available:")
+ for s_type, info in SCREENPLAY_LENGTHS.items():
+ logger.info(f" - {s_type}: {info['description']}")
+ logger.info(f"Genres: {', '.join(GENRE_TEMPLATES.keys())}")
+
+ if BRAVE_SEARCH_API_KEY:
+ logger.info("Web search enabled for market research.")
+ else:
+ logger.warning("Web search disabled.")
+
+ logger.info("=" * 60)
+
+ # Initialize database
+ logger.info("Initializing database...")
+ ScreenplayDatabase.init_db()
+ logger.info("Database initialization complete.")
+
+ # Create and launch interface
+ interface = create_interface()
+
+ interface.launch(
+ server_name="0.0.0.0",
+ server_port=7860,
+ share=False,
+ debug=True
+ )
\ No newline at end of file
- 
- 
-