Update app.py

app.py

@@ -4,7 +4,7 @@ from pydantic import BaseModel
 import time
 import os
 import base64
-import uuid
+import uuidfrom datetime import datetime, timedelta, timezone
 from google import genai
 from google.genai import types
 from typing import List
@@ -39,7 +39,7 @@ user = []
 sessions = {}  # Format: {uid: {conv_id: [history_list]}}
 
 
-user_details = [] #[[name, gender, nationality, model name, model gender]]
+user_data = [] #[[name, gender, nationality, model name, model gender]]
 user_inst = [] #[[system prompt, About user, demanded tone, custum instruction]]
 
 
@@ -60,6 +60,40 @@ class VerifyRequest(BaseModel):
     uid: str
     idToken: str
 
+class UpdateSystemRequest(BaseModel):
+    category: str
+    content: str
+
+def update_details():
+    user_data.append(['','','','',''])
+    user_inst.append(['','','',''])
+    
+@app.post("/update/system/userInfo")
+async def update_Details(info: UpdateSystemRequest, token: str = Depends(verify_access_token)):
+    i = Tokens.index(token)
+    cat = info.category
+    val = info.content
+    if cat == 'name':
+        user_data[i][0] = val
+    if cat == 'gender':
+        user_data[i][1] = val
+    if cat == 'country':
+        user_data[i][2] = val
+    if cat == 'model_name':
+        user_data[i][3] = val
+    if cat == 'model_gender':
+        user_data[i][4] = val
+    if cat == 'system_prompt':
+        user_inst[i][0] =val
+    if cat == 'about_user':
+        user_inst[i][1] =val
+    if cat == 'tone':
+        user_inst[i][2] =val
+    if cat == 'custum_instruction':
+        user_inst[i][3] =val
+    else:
+        raise HTTPException(status_code=401, detail="Invalid Data.")
+
 
 @app.post("/api/verify")
 def check_token(data: VerifyRequest):
@@ -73,6 +107,7 @@ def check_token(data: VerifyRequest):
             user.append(uid)
             Tokens.append(new_token)
             sessions[uid] = {}
+            update_details()
         else:
             idx = user.index(uid)
             Tokens[idx] = new_token
@@ -114,6 +149,11 @@ def getConvId():
     u = uuid.uuid4().bytes[:12]
     return base64.urlsafe_b64encode(u).rstrip(b'=').decode()
 
+def currentTime():
+    # IST is UTC + 5:30
+    ist_offset = timezone(timedelta(hours=5, minutes=30))
+    current_time = datetime.now(ist_offset)
+    return current_time.strftime("%I:%M %p")
 
 def verify_access_token(auth: HTTPAuthorizationCredentials = Depends(security)):
     token = auth.credentials
@@ -132,7 +172,7 @@ async def handleNewConv(new_conv: NewConv, token: str = Depends(verify_access_to
     convs = sessions.get(user[i])
     history = [
         types.Content(role='model', parts=[types.Part(text=getSystemPrompt())]),
-        types.Content(role='user', parts=[types.Part(text=new_conv.prompt)])
+        types.Content(role='user', parts=[types.Part(text= f'time {currentTime()}\n{new_conv.prompt)}'])
     ]
     text = call_gemini(history)
     if text:
@@ -158,7 +198,7 @@ async def handleChat(chat_request: ChatRequest, token: str = Depends(verify_acce
         types.Content(role=k[0], parts=[types.Part(text=k[1])]) 
         for k in raw_history
     ]
-    history.append(types.Content(role='user', parts=[types.Part(text=chat_request.prompt)]))
+    history.append(types.Content(role='user', parts=[types.Part(text= f'time {currentTime()}\n{chat_request.prompt)}']))
     # Generate response
     text = call_gemini(history)
     if text: