Update app.py

app.py

@@ -1742,17 +1742,21 @@ def format_stages_html(stages: List[Dict[str, str]], language: str, auto_scroll:
 
 def get_active_sessions(language: str) -> List[Tuple[str, str, str]]:
     """Get list of active sessions"""
-    sessions = NovelDatabase.get_active_sessions()
-    
-    choices = []
-    for session in sessions:
-        created = datetime.fromisoformat(session['created_at'])
-        date_str = created.strftime("%Y-%m-%d %H:%M")
-        query_preview = session['user_query'][:50] + "..." if len(session['user_query']) > 50 else session['user_query']
-        label = f"[{date_str}] {query_preview} (Stage {session['current_stage']})"
-        choices.append((label, session['session_id']))
-    
-    return choices
+    try:
+        sessions = NovelDatabase.get_active_sessions()
+        
+        choices = []
+        for session in sessions:
+            created = datetime.fromisoformat(session['created_at'])
+            date_str = created.strftime("%Y-%m-%d %H:%M")
+            query_preview = session['user_query'][:50] + "..." if len(session['user_query']) > 50 else session['user_query']
+            label = f"[{date_str}] {query_preview} (Stage {session['current_stage']})"
+            choices.append((label, session['session_id']))
+        
+        return choices
+    except Exception as e:
+        logger.error(f"Error getting active sessions: {str(e)}")
+        return []
 
 def resume_session(session_id: str, language: str) -> Generator[Tuple[str, str, str], None, None]:
     """Resume an existing session"""
@@ -2060,8 +2064,12 @@ def create_interface():
             return process, novel, status, novel
         
         def refresh_sessions():
-            sessions = get_active_sessions("English")
-            return gr.update(choices=sessions)
+            try:
+                sessions = get_active_sessions("English")
+                return gr.update(choices=sessions)
+            except Exception as e:
+                logger.error(f"Error refreshing sessions: {str(e)}")
+                return gr.update(choices=[])
         
         submit_btn.click(
             fn=process_query,
@@ -2127,6 +2135,11 @@ def create_interface():
 if __name__ == "__main__":
     logger.info("Starting SOMA Novel Writing System...")
     
+    # Initialize database on startup
+    logger.info("Initializing database...")
+    NovelDatabase.init_db()
+    logger.info("Database initialized successfully.")
+    
     interface = create_interface()
     
     interface.launch(