Update app.py

app.py

@@ -2980,106 +2980,106 @@ def download_screenplay(screenplay_text: str, format_type: str, title: str,
 
 # Create Gradio interface
 def create_interface():
-   """Create Gradio interface for screenplay generation"""
-   
-   css = """
-   .main-header {
-       text-align: center;
-       margin-bottom: 2rem;
-       padding: 2rem;
-       background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%);
-       border-radius: 10px;
-       color: white;
-   }
-   
-   .header-title {
-       font-size: 3rem;
-       margin-bottom: 1rem;
-       background: linear-gradient(45deg, #f39c12, #e74c3c);
-       -webkit-background-clip: text;
-       -webkit-text-fill-color: transparent;
-   }
-   
-   .header-description {
-       font-size: 1.1rem;
-       opacity: 0.9;
-       line-height: 1.6;
-   }
-   
-   .type-selector {
-       display: flex;
-       gap: 1rem;
-       margin: 1rem 0;
-   }
-   
-   .type-card {
-       flex: 1;
-       padding: 1rem;
-       border: 2px solid #ddd;
-       border-radius: 8px;
-       cursor: pointer;
-       transition: all 0.3s;
-   }
-   
-   .type-card:hover {
-       border-color: #f39c12;
-       transform: translateY(-2px);
-   }
-   
-   .type-card.selected {
-       border-color: #e74c3c;
-       background: #fff5f5;
-   }
-   
-   #stages-display {
-       max-height: 600px;
-       overflow-y: auto;
-       padding: 1rem;
-       background: #f8f9fa;
-       border-radius: 8px;
-   }
-   
-   #screenplay-output {
-       font-family: 'Courier New', monospace;
-       white-space: pre-wrap;
-       background: white;
-       padding: 2rem;
-       border: 1px solid #ddd;
-       border-radius: 8px;
-       max-height: 800px;
-       overflow-y: auto;
-   }
-   
-   .genre-grid {
-       display: grid;
-       grid-template-columns: repeat(auto-fit, minmax(150px, 1fr));
-       gap: 0.5rem;
-       margin: 1rem 0;
-   }
-   
-   .genre-btn {
-       padding: 0.75rem;
-       border: 2px solid #e0e0e0;
-       background: white;
-       border-radius: 8px;
-       cursor: pointer;
-       transition: all 0.3s;
-       text-align: center;
-   }
-   
-   .genre-btn:hover {
-       border-color: #f39c12;
-       background: #fffbf0;
-   }
-   
-   .genre-btn.selected {
-       border-color: #e74c3c;
-       background: #fff5f5;
-       font-weight: bold;
-   }
-   """
-   
-   with gr.Blocks(theme=gr.themes.Soft(), css=css, title="Screenplay Generator") as interface:
+    """Create Gradio interface for screenplay generation"""
+    
+    css = """
+    .main-header {
+        text-align: center;
+        margin-bottom: 2rem;
+        padding: 2rem;
+        background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%);
+        border-radius: 10px;
+        color: white;
+    }
+    
+    .header-title {
+        font-size: 3rem;
+        margin-bottom: 1rem;
+        background: linear-gradient(45deg, #f39c12, #e74c3c);
+        -webkit-background-clip: text;
+        -webkit-text-fill-color: transparent;
+    }
+    
+    .header-description {
+        font-size: 1.1rem;
+        opacity: 0.9;
+        line-height: 1.6;
+    }
+    
+    .type-selector {
+        display: flex;
+        gap: 1rem;
+        margin: 1rem 0;
+    }
+    
+    .type-card {
+        flex: 1;
+        padding: 1rem;
+        border: 2px solid #ddd;
+        border-radius: 8px;
+        cursor: pointer;
+        transition: all 0.3s;
+    }
+    
+    .type-card:hover {
+        border-color: #f39c12;
+        transform: translateY(-2px);
+    }
+    
+    .type-card.selected {
+        border-color: #e74c3c;
+        background: #fff5f5;
+    }
+    
+    #stages-display {
+        max-height: 600px;
+        overflow-y: auto;
+        padding: 1rem;
+        background: #f8f9fa;
+        border-radius: 8px;
+    }
+    
+    #screenplay-output {
+        font-family: 'Courier New', monospace;
+        white-space: pre-wrap;
+        background: white;
+        padding: 2rem;
+        border: 1px solid #ddd;
+        border-radius: 8px;
+        max-height: 800px;
+        overflow-y: auto;
+    }
+    
+    .genre-grid {
+        display: grid;
+        grid-template-columns: repeat(auto-fit, minmax(150px, 1fr));
+        gap: 0.5rem;
+        margin: 1rem 0;
+    }
+    
+    .genre-btn {
+        padding: 0.75rem;
+        border: 2px solid #e0e0e0;
+        background: white;
+        border-radius: 8px;
+        cursor: pointer;
+        transition: all 0.3s;
+        text-align: center;
+    }
+    
+    .genre-btn:hover {
+        border-color: #f39c12;
+        background: #fffbf0;
+    }
+    
+    .genre-btn.selected {
+        border-color: #e74c3c;
+        background: #fff5f5;
+        font-weight: bold;
+    }
+    """
+    
+    with gr.Blocks(theme=gr.themes.Soft(), css=css, title="Screenplay Generator") as interface:
         gr.HTML("""
         <style>
             .badges-container {
@@ -3138,217 +3138,217 @@ def create_interface():
             </p>
         </div>
         """)
-       # State management
-       current_session_id = gr.State(None)
-       
-       with gr.Tabs():
-           # Main Writing Tab
-           with gr.Tab("✍️ Write Screenplay"):
-               with gr.Row():
-                   with gr.Column(scale=3):
-                       query_input = gr.Textbox(
-                           label="Screenplay Concept",
-                           placeholder="""Describe your screenplay idea. For example:
+        
+        # State management
+        current_session_id = gr.State(None)
+        
+        with gr.Tabs():
+            # Main Writing Tab
+            with gr.Tab("✍️ Write Screenplay"):
+                with gr.Row():
+                    with gr.Column(scale=3):
+                        query_input = gr.Textbox(
+                            label="Screenplay Concept",
+                            placeholder="""Describe your screenplay idea. For example:
 - A detective with memory loss must solve their own attempted murder
 - Two rival food truck owners forced to work together to save the city food festival  
 - A space station AI develops consciousness during a critical mission
 - A family reunion turns into a murder mystery during a hurricane
 
 The more specific your concept, the better the screenplay will be tailored to your vision.""",
-                           lines=6
-                       )
-                   
-                   with gr.Column(scale=1):
-                       screenplay_type = gr.Radio(
-                           choices=list(SCREENPLAY_LENGTHS.keys()),
-                           value="movie",
-                           label="Screenplay Type",
-                           info="Choose your format"
-                       )
-                       
-                       genre_select = gr.Dropdown(
-                           choices=list(GENRE_TEMPLATES.keys()),
-                           value="drama",
-                           label="Primary Genre",
-                           info="Select main genre"
-                       )
-                       
-                       language_select = gr.Radio(
-                           choices=["English", "Korean"],
-                           value="English",
-                           label="Language"
-                       )
-               
-               with gr.Row():
-                   random_btn = gr.Button("🎲 Random Concept", scale=1)
-                   clear_btn = gr.Button("🗑️ Clear", scale=1)
-                   submit_btn = gr.Button("🎬 Start Writing", variant="primary", scale=2)
-               
-               status_text = gr.Textbox(
-                   label="Status",
-                   interactive=False,
-                   value="Ready to create your screenplay"
-               )
-               
-               # Session management
-               with gr.Group():
-                   gr.Markdown("### 📁 Saved Projects")
-                   with gr.Row():
-                       session_dropdown = gr.Dropdown(
-                           label="Active Sessions",
-                           choices=[],
-                           interactive=True,
-                           scale=3
-                       )
-                       refresh_btn = gr.Button("🔄", scale=1)
-                       resume_btn = gr.Button("📂 Load", scale=1)
-               
-               # Output displays
-               with gr.Row():
-                   with gr.Column():
-                       with gr.Tab("🎭 Writing Progress"):
-                           stages_display = gr.Markdown(
-                               value="*Your screenplay journey will unfold here...*",
-                               elem_id="stages-display"
-                           )
-                       
-                       with gr.Tab("📄 Screenplay"):
-                           screenplay_output = gr.Markdown(
-                               value="*Your formatted screenplay will appear here...*",
-                               elem_id="screenplay-output"
-                           )
-                           
-                           with gr.Row():
-                               format_select = gr.Radio(
-                                   choices=["PDF", "FDX", "FOUNTAIN", "TXT"],
-                                   value="PDF",
-                                   label="Export Format"
-                               )
-                               download_btn = gr.Button("📥 Download Screenplay", variant="secondary")
-                           
-                           download_file = gr.File(
-                               label="Download",
-                               visible=False
-                           )
-               
-               # Examples
-               gr.Examples(
-                   examples=[
-                       ["A burned-out teacher discovers her students are being replaced by AI duplicates"],
-                       ["Two funeral home employees accidentally release a ghost who helps them solve murders"],
-                       ["A time-loop forces a wedding planner to relive the worst wedding until they find true love"],
-                       ["An astronaut returns to Earth to find everyone has forgotten space exists"],
-                       ["A support group for reformed villains must save the city when heroes disappear"],
-                       ["A food critic loses their sense of taste and teams up with a street food vendor"]
-                   ],
-                   inputs=query_input,
-                   label="💡 Example Concepts"
-               )
-           
-           # Screenplay Library Tab
-           with gr.Tab("📚 Concept Library"):
-               gr.Markdown("""
-               ### 🎲 Random Screenplay Concepts
-               
-               Browse through AI-generated screenplay concepts. Each concept includes a title, logline, and brief setup.
-               """)
-               
-               library_display = gr.HTML(
-                   value="<p>Library feature coming soon...</p>"
-               )
-       
-       # Event handlers
-       def handle_submit(query, s_type, genre, lang, session_id):
-           if not query:
-               yield "", "", "❌ Please enter a concept", session_id
-               return
-           
-           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
+                            lines=6
+                        )
+                    
+                    with gr.Column(scale=1):
+                        screenplay_type = gr.Radio(
+                            choices=list(SCREENPLAY_LENGTHS.keys()),
+                            value="movie",
+                            label="Screenplay Type",
+                            info="Choose your format"
+                        )
+                        
+                        genre_select = gr.Dropdown(
+                            choices=list(GENRE_TEMPLATES.keys()),
+                            value="drama",
+                            label="Primary Genre",
+                            info="Select main genre"
+                        )
+                        
+                        language_select = gr.Radio(
+                            choices=["English", "Korean"],
+                            value="English",
+                            label="Language"
+                        )
+                
+                with gr.Row():
+                    random_btn = gr.Button("🎲 Random Concept", scale=1)
+                    clear_btn = gr.Button("🗑️ Clear", scale=1)
+                    submit_btn = gr.Button("🎬 Start Writing", variant="primary", scale=2)
+                
+                status_text = gr.Textbox(
+                    label="Status",
+                    interactive=False,
+                    value="Ready to create your screenplay"
+                )
+                
+                # Session management
+                with gr.Group():
+                    gr.Markdown("### 📁 Saved Projects")
+                    with gr.Row():
+                        session_dropdown = gr.Dropdown(
+                            label="Active Sessions",
+                            choices=[],
+                            interactive=True,
+                            scale=3
+                        )
+                        refresh_btn = gr.Button("🔄", scale=1)
+                        resume_btn = gr.Button("📂 Load", scale=1)
+                
+                # Output displays
+                with gr.Row():
+                    with gr.Column():
+                        with gr.Tab("🎭 Writing Progress"):
+                            stages_display = gr.Markdown(
+                                value="*Your screenplay journey will unfold here...*",
+                                elem_id="stages-display"
+                            )
+                        
+                        with gr.Tab("📄 Screenplay"):
+                            screenplay_output = gr.Markdown(
+                                value="*Your formatted screenplay will appear here...*",
+                                elem_id="screenplay-output"
+                            )
+                            
+                            with gr.Row():
+                                format_select = gr.Radio(
+                                    choices=["PDF", "FDX", "FOUNTAIN", "TXT"],
+                                    value="PDF",
+                                    label="Export Format"
+                                )
+                                download_btn = gr.Button("📥 Download Screenplay", variant="secondary")
+                            
+                            download_file = gr.File(
+                                label="Download",
+                                visible=False
+                            )
+                
+                # Examples
+                gr.Examples(
+                    examples=[
+                        ["A burned-out teacher discovers her students are being replaced by AI duplicates"],
+                        ["Two funeral home employees accidentally release a ghost who helps them solve murders"],
+                        ["A time-loop forces a wedding planner to relive the worst wedding until they find true love"],
+                        ["An astronaut returns to Earth to find everyone has forgotten space exists"],
+                        ["A support group for reformed villains must save the city when heroes disappear"],
+                        ["A food critic loses their sense of taste and teams up with a street food vendor"]
+                    ],
+                    inputs=query_input,
+                    label="💡 Example Concepts"
+                )
+            
+            # Screenplay Library Tab
+            with gr.Tab("📚 Concept Library"):
+                gr.Markdown("""
+                ### 🎲 Random Screenplay Concepts
+                
+                Browse through AI-generated screenplay concepts. Each concept includes a title, logline, and brief setup.
+                """)
+                
+                library_display = gr.HTML(
+                    value="<p>Library feature coming soon...</p>"
+                )
+        
+        # Event handlers
+        def handle_submit(query, s_type, genre, lang, session_id):
+            if not query:
+                yield "", "", "❌ Please enter a concept", session_id
+                return
+            
+            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
 
 # Main function
 if __name__ == "__main__":
-   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
-   )
-    
+    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
+    )