Update utils.py

utils.py

@@ -237,30 +237,44 @@ def generate_follow_up_questions(idea_form: IdeaForm, current_stage: int) -> Lis
 
     return questions
 
-def save_idea_to_database(form: IdeaForm, db: Session) -> int:
-    """
-    Saves the IdeaForm to the SQLite database.
-    Returns the ID of the inserted row.
-    """
-    db_idea = InnovativeIdea(
-        idea_name=form.idea_name,
-        idea_overview=form.idea_overview,
-        problem_solution=form.problem_solution,
-        target_customers=form.target_customers,
-        market_value=form.market_value,
-        existing_solutions=form.existing_solutions,
-        unique_value=form.unique_value,
-        technical_challenges=form.technical_challenges,
-        legal_barriers=form.legal_barriers,
-        data_dependencies=form.data_dependencies,
-        team_roles=','.join(form.team_roles) if form.team_roles else None,
-        timeline=form.timeline,
-        additional_info=form.additional_info
-    )
-    db.add(db_idea)
-    db.commit()
-    db.refresh(db_idea)
-    return db_idea.id
+def update_idea_in_database(idea_id: int, form: IdeaForm, db: Session) -> bool:
+    try:
+        db_idea = db.query(InnovativeIdea).filter(InnovativeIdea.id == idea_id).first()
+        if db_idea:
+            for field, value in form.dict(exclude_unset=True).items():
+                setattr(db_idea, field, value)
+            db.commit()
+            return True
+    except Exception as e:
+        db.rollback()
+        logging.error(f"Error updating idea in database: {str(e)}")
+    return False
+
+def save_idea_to_database(form: IdeaForm, db: Session) -> Optional[int]:
+    try:
+        db_idea = InnovativeIdea(
+            idea_name=form.idea_name,
+            idea_overview=form.idea_overview,
+            problem_solution=form.problem_solution,
+            target_customers=form.target_customers,
+            market_value=form.market_value,
+            existing_solutions=form.existing_solutions,
+            unique_value=form.unique_value,
+            technical_challenges=form.technical_challenges,
+            legal_barriers=form.legal_barriers,
+            data_dependencies=form.data_dependencies,
+            team_roles=','.join(form.team_roles) if form.team_roles else None,
+            timeline=form.timeline,
+            additional_info=form.additional_info
+        )
+        db.add(db_idea)
+        db.commit()
+        db.refresh(db_idea)
+        return db_idea.id
+    except Exception as e:
+        db.rollback()
+        logging.error(f"Error saving idea to database: {str(e)}")
+    return None
 
 def load_idea_from_database(idea_id: int, db: Session) -> Optional[IdeaForm]:
     """
@@ -285,18 +299,6 @@ def load_idea_from_database(idea_id: int, db: Session) -> Optional[IdeaForm]:
         )
     return None
 
-def update_idea_in_database(idea_id: int, form: IdeaForm, db: Session) -> bool:
-    """
-    Updates an existing idea in the database.
-    """
-    db_idea = db.query(InnovativeIdea).filter(InnovativeIdea.id == idea_id).first()
-    if db_idea:
-        for field, value in form.dict(exclude_unset=True).items():
-            setattr(db_idea, field, value)
-        db.commit()
-        return True
-    return False
-
 def extract_search_query(message: str) -> Optional[str]:
     """
     Extract a search query from a user message if it starts with '@'.