Update api.py

api.py

@@ -55,7 +55,7 @@ class SpecificityEvaluationResponse(BaseModel):
     specificity: SpecificityScore
 
 class ProblemDescriptionRequest(BaseModel):
-    description: str
+    descriptions: List[str]
     technical_topic: str
 
 class ProblemDescriptionResponse(BaseModel):
@@ -140,6 +140,63 @@ app = FastAPI(
     lifespan=lifespan # Use the lifespan context manager
 )
 
+# --- Helper Functions for Prompt Building ---
+
+def format_ki_descriptions(descriptions: list[str]) -> str:
+  formatted_descriptions = ""
+  for ind in range(len(sample)):
+    ki_number = ind + 1
+    formatted_description = f"Here is the description of the key issue {ki_number}: <description{ki_number}>{descriptions[ind]}<description{ki_number}>"
+    if ind == len(sample) - 1:
+      formatted_descriptions += formatted_description
+    else:
+      formatted_descriptions += formatted_description + "\n"
+  return formatted_descriptions
+
+def build_prompt(descriptions: list[str], technical_topic: str) -> str | None:
+  prompt = None
+  if len(descriptions) == 1:
+    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>{descriptions[0]}</description>
+
+This key issue is part of the following technical topic: <technical_topic>{technical_topic}</technical_topic>
+
+As the output, I only want you to provide the problem description found, nothing else.
+
+Here are examples of problem descriptions that you could create, it shows the level of specificity that you should aim for:
+
+Example 1: 'I am working on enhancing security in 4G and 5G telecommunications networks, specifically in the area of network slicing. My goal is to address vulnerabilities in the isolation of slices and improve the integrity of data transmission across different network slices, ensuring they are properly protected against unauthorized access and attacks.'
+Example 2: 'I am working on improving user authentication in the context of 3GPP (3rd Generation Partnership Project) networks. Specifically, I need to solve issues related to the security and efficiency of the authentication process in the 5G network architecture, focusing on the use of cryptographic keys and identity management.'
+
+As far as possible, avoid using acronyms in the problem description.
+Try to be about the same length as the examples if possible."""
+
+  elif len(descriptions) > 1:
+    formatted_descriptions = format_ki_descriptions(descriptions)
+    prompt = f"""I want you to create a problem description using several key issues, each explained with a detailed description.
+
+{formatted_descriptions}
+
+These key issues are part of the following technical topic: <technical_topic>{technical_topic}<technical_topic>
+
+As the output, I only want you to provide the problem description found, nothing else.
+
+Here are examples of problem descriptions that you could create, it shows the level of specificity that you should aim for:
+
+Example 1: 'I am working on enhancing security in 4G and 5G telecommunications networks, specifically in the area of network slicing.
+My goal is to address vulnerabilities in the isolation of slices and improve the integrity of data transmission across different network slices,
+ensuring they are properly protected against unauthorized access and attacks.'
+
+Example 2: 'I am working on improving user authentication in the context of 3GPP (3rd Generation Partnership Project) networks.
+Specifically, I need to solve issues related to the security and efficiency of the authentication process in the 5G network architecture,
+focusing on the use of cryptographic keys and identity management.'
+
+As far as possible, avoid using acronyms in the problem description.
+When creating the problem description that combines the topics of the key issues, be as intelligible as possible and ensure that the mix of key issues makes sense.
+The description of the problem should not exceed 100 words or so."""
+
+  return prompt
+
 # --- API Endpoint ---
 # API state check route
 @app.get("/")
@@ -349,34 +406,33 @@ async def evaluation(request: SpecificityEvaluationRequest):
 @app.post("/create-problem-description", response_model=ProblemDescriptionResponse)
 async def gen_problem_description(request: ProblemDescriptionRequest):
     """
-    Generates a problem description using Gemini based on a detailed description
-    and technical topic.
+    Generates a problem description using Gemini based on one or more detailed
+    descriptions and a technical topic.
     """
-    description = request.description
+    # Check if Gemini library was imported and configured
+    if not genai or not genai.is_configured():
+        logger.error("Gemini client is not available or configured.")
+        raise HTTPException(status_code=503, detail="Service Unavailable: Gemini client not configured")
+
+    descriptions = request.descriptions
     technical_topic = request.technical_topic
 
-    if not description or not technical_topic:
-        raise HTTPException(status_code=400, detail="Missing required fields: description or technical_topic.")
+    # Validate input: need at least one description and a topic
+    if not descriptions: # Check if the list is empty
+        raise HTTPException(status_code=400, detail="Field 'descriptions' cannot be empty.")
+    if not technical_topic:
+        raise HTTPException(status_code=400, detail="Missing required field: technical_topic.")
 
-    logger.info("Received request for problem description generation.")
-    logger.debug(f"Topic: {technical_topic}") # Avoid logging full description unless necessary
+    logger.info(f"Received request for problem description generation with {len(descriptions)} description(s).")
+    logger.debug(f"Topic: {technical_topic}")
 
-    # --- Construct Prompt ---
-    prompt = f"""I want you to create a problem description using a key issue explained in a detailed description.
+    # --- Build Prompt using helper function ---
+    prompt = build_prompt(descriptions, technical_topic)
 
-    Here is the description of the key issue: <description>{description}</description>
-    
-    This key issue is part of the following technical topic: <technical_topic>{technical_topic}</technical_topic>
-    
-    As the output, I only want you to provide the problem description found, nothing else.
-    
-    Here are examples of problem descriptions that you could create, it shows the level of specificity that you should aim for:
-    
-    Example 1: 'I am working on enhancing security in 4G and 5G telecommunications networks, specifically in the area of network slicing. My goal is to address vulnerabilities in the isolation of slices and improve the integrity of data transmission across different network slices, ensuring they are properly protected against unauthorized access and attacks.'
-    Example 2: 'I am working on improving user authentication in the context of 3GPP (3rd Generation Partnership Project) networks. Specifically, I need to solve issues related to the security and efficiency of the authentication process in the 5G network architecture, focusing on the use of cryptographic keys and identity management.'
-    
-    As far as possible, avoid using acronyms in the problem description.
-    Try to be about the same length as the examples if possible."""
+    if prompt is None:
+        # This case should theoretically be caught by the validation above, but good practice to check
+        logger.error("Failed to build prompt (likely empty description list).")
+        raise HTTPException(status_code=500, detail="Internal error: Could not build prompt.")
 
     try:
         logger.info("Calling Gemini API to generate problem description...")
@@ -402,7 +458,6 @@ async def gen_problem_description(request: ProblemDescriptionRequest):
     # --- Return Result ---
     # Return as JSON using the Pydantic model
     return ProblemDescriptionResponse(problem_description=result_text)
-
     # Alternative: Return as plain text
     # return PlainTextResponse(content=result_text)