Update tp_final.py

tp_final.py

@@ -216,12 +216,14 @@ except Exception as e:
 
 # --- Pydantic Models for Request/Response ---
 class AnalysisRequest(BaseModel):
-    username: str = "sameer"  # Default as per your code
+    user_id: str
+    username: str   # Default as per your code
     user_journal: str
     # Add other fields here if you need to pass user_journal dynamically later
 
 class AnalysisResponse(BaseModel):
     status: str
+    user_id: str
     username: str
     guidelines: Dict[str, Any]
     recommended_tasks: List[str]
@@ -235,14 +237,7 @@ async def cron_test():
 
 @app.post("/analyze-tasks", response_model=AnalysisResponse)
 def generate_analysis(request: AnalysisRequest):
-    """
-    Executes the analysis pipeline: 
-    1. Identifies direct/indirect problems via Groq.
-    2. Generates guidelines via Groq.
-    3. Matches problems to Yoga tasks using TF-IDF/Cosine Similarity.
-    4. Saves to MongoDB.
-    """
-    
+
     example = get_file_data("example.json")
     direcindirecprompt = f'''
             I'll be giving you a journal entry written by a user. Your task is to extract out problems being faced by the user in the following format:
@@ -421,7 +416,8 @@ def generate_analysis(request: AnalysisRequest):
             guidelines_obj = {"raw_text": guidegroq_response}
 
         chat_document = {
-            "user_id": request.username,
+            "user_id": request.user_id,
+            "username": request.username,
             "timestamp": datetime.now(),
             "problemdesc": direcindirecgroq_response, # Storing raw response string as per original code
             "guidelines": guidegroq_response,       # Storing raw response string as per original code