Update file_utils.py

file_utils.py

@@ -30,6 +30,11 @@ class UserFileEntry(SQLModel, table=True):
     file_path: str
     timestamp: str
 
+class UserProfile(SQLModel, table=True):
+    id: Optional[int] = Field(default=None, primary_key=True)
+    nickname: str
+    timestamp: str
+
 SQLModel.metadata.create_all(engine)
 
 # === File Persistence ===
@@ -71,6 +76,21 @@ def fetch_study_guides(user):
     session.close()
     return results
 
+def save_nickname_to_db(nickname):
+    session = Session(engine)
+    existing = session.exec(select(UserProfile).where(UserProfile.nickname == nickname)).first()
+    if not existing:
+        entry = UserProfile(nickname=nickname, timestamp=datetime.now().strftime("%Y-%m-%d %H:%M"))
+        session.add(entry)
+        session.commit()
+    session.close()
+
+def load_latest_nickname():
+    session = Session(engine)
+    result = session.exec(select(UserProfile).order_by(UserProfile.id.desc())).first()
+    session.close()
+    return result.nickname if result else ""
+
 # === LLM Core ===
 def call_llm(prompt=None, system_message="You are a helpful assistant.", messages=None):
     max_safe_chars = 28000