Update api.py

api.py

@@ -73,12 +73,18 @@ class FormattedKeyIssue(BaseModel):
 class KeyIssueFormatResponse(BaseModel):
     key_issues: List[FormattedKeyIssue]
 
+class KeyIssueInput(BaseModel):
+    """Represents a single key issue with its description and challenges."""
+    description: str
+    challenges: List[str]
+
 class CreateSeveralProbDescRequest(BaseModel):
-    descriptions: Dict[int, str]
-    challenges: Dict[int, List[str]]
+    """Request body for creating multiple problem descriptions."""
+    keys_issues: List[KeyIssueInput] # Changed field name and type
     technical_topic: str
 
 class CreateSeveralProbDescResponse(BaseModel):
+    """Response body containing the list of generated problem descriptions."""
     problem_descriptions: List[str]
 
 # --- Global Variables / State ---
@@ -493,8 +499,7 @@ async def create_several_probdesc(request: CreateSeveralProbDescRequest):
         logger.error("Gemini client is not available or configured.")
         raise HTTPException(status_code=503, detail="Service Unavailable: Gemini client not configured")
 
-    descriptions_map = request.descriptions
-    challenges_map = request.challenges
+    key_issues_list = request.keys_issues
     technical_topic = request.technical_topic
 
     # Basic input validation
@@ -509,24 +514,25 @@ async def create_several_probdesc(request: CreateSeveralProbDescRequest):
     generated_descriptions = []
 
     # Iterate through challenges to create tasks
-    for key_issue_id, list_of_challenges in challenges_map.items():
-        # Get the corresponding description
-        description = descriptions_map.get(key_issue_id)
-        successful_count = 0
-        failed_count = 0
+    for key_issue in key_issues_list:
+        description = key_issue.description
+        list_of_challenges = key_issue.challenges
+
         if description is None:
             logger.warning(f"No description found for key_issue_id {key_issue_id}. Skipping its challenges.")
             continue # Skip challenges for this ID if description is missing
 
         if not list_of_challenges:
-             logger.warning(f"Empty challenge list for key_issue_id {key_issue_id}. Skipping.")
-             continue
+            logger.warning(f"Empty challenge list for key_issue_id {key_issue_id}. Skipping.")
+            continue
 
-        # Create a task for each challenge associated with this description
         for ind, challenge in enumerate(list_of_challenges):
+            successful_count = 0
+            failed_count = 0
+            # Create a task for each challenge associated with this description
             if not challenge: # Skip empty challenges
-                 logger.warning(f"Skipping empty challenge string for key_issue_id {key_issue_id}.")
-                 continue
+                logger.warning(f"Skipping empty challenge string for key_issue_id {key_issue_id}.")
+                continue
             else:
                 prompt = f"""I want you to create a problem description using a key issue explained in a detailed description.
 Here is the description of the key issue: <description>{description}</description>
@@ -558,9 +564,10 @@ Try to be about the same length as the examples if possible."""
                     failed_count += 1
                     logger.error(f"Error calling Gemini API for problem description {ind+1}: {e}", exc_info=True)
                     # Check for specific Gemini API errors if possible
-                    raise HTTPException(status_code=502, detail=f"Failed to generate problem description using LLM: {e}")
+                    raise HTTPException(status_code=502, detail=f"Failed to generate problem description {ind+1} using LLM: {e}")
 
         logger.info(f"Successfully generated descriptions: {successful_count}/{len(list_of_challenges)} ")
+        
 
     # --- Return Result ---
     return CreateSeveralProbDescResponse(problem_descriptions=generated_descriptions)