Complete Activity 1

Author: petern148

app/main.py

@@ -44,14 +44,12 @@ def _get_openai_client():
     """Lazily initialize the Azure OpenAI client."""
     global _openai_client
     if _openai_client is None:
-        # TODO: Uncomment and configure
-        #   from openai import AzureOpenAI
-        #   _openai_client = AzureOpenAI(
-        #       azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
-        #       api_key=os.environ["AZURE_OPENAI_API_KEY"],
-        #       api_version="2024-10-21",
-        #   )
-        raise NotImplementedError("Configure the Azure OpenAI client")
+        from openai import AzureOpenAI
+        _openai_client = AzureOpenAI(
+            azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
+            api_key=os.environ["AZURE_OPENAI_API_KEY"],
+            api_version="2024-10-21",
+        )
     return _openai_client
 
 
@@ -61,14 +59,12 @@ def _get_content_safety_client():
     if _content_safety_client is None:
         # NOTE: The Content Safety SDK handles API versioning internally --
         # no api_version parameter is needed (unlike the OpenAI SDK).
-        # TODO: Uncomment and configure
-        #   from azure.ai.contentsafety import ContentSafetyClient
-        #   from azure.core.credentials import AzureKeyCredential
-        #   _content_safety_client = ContentSafetyClient(
-        #       endpoint=os.environ["AZURE_CONTENT_SAFETY_ENDPOINT"],
-        #       credential=AzureKeyCredential(os.environ["AZURE_CONTENT_SAFETY_KEY"]),
-        #   )
-        raise NotImplementedError("Configure the Content Safety client")
+        from azure.ai.contentsafety import ContentSafetyClient
+        from azure.core.credentials import AzureKeyCredential
+        _content_safety_client = ContentSafetyClient(
+            endpoint=os.environ["AZURE_CONTENT_SAFETY_ENDPOINT"],
+            credential=AzureKeyCredential(os.environ["AZURE_CONTENT_SAFETY_KEY"]),
+        )
     return _content_safety_client
 
 
@@ -78,14 +74,12 @@ def _get_language_client():
     if _language_client is None:
         # NOTE: The Language SDK handles API versioning internally --
         # no api_version parameter is needed (unlike the OpenAI SDK).
-        # TODO: Uncomment and configure
-        #   from azure.ai.textanalytics import TextAnalyticsClient
-        #   from azure.core.credentials import AzureKeyCredential
-        #   _language_client = TextAnalyticsClient(
-        #       endpoint=os.environ["AZURE_AI_LANGUAGE_ENDPOINT"],
-        #       credential=AzureKeyCredential(os.environ["AZURE_AI_LANGUAGE_KEY"]),
-        #   )
-        raise NotImplementedError("Configure the AI Language client")
+        from azure.ai.textanalytics import TextAnalyticsClient
+        from azure.core.credentials import AzureKeyCredential
+        _language_client = TextAnalyticsClient(
+            endpoint=os.environ["AZURE_AI_LANGUAGE_ENDPOINT"],
+            credential=AzureKeyCredential(os.environ["AZURE_AI_LANGUAGE_KEY"]),
+        )
     return _language_client
 
 
@@ -100,7 +94,7 @@ def classify_311_request(request_text: str) -> dict:
 
     Returns:
         dict with keys: category, confidence, reasoning
-    """
+    
     # TODO: Step 1.1 - Get the OpenAI client
     # TODO: Step 1.2 - Call client.chat.completions.create() with:
     #   model=os.environ.get("AZURE_OPENAI_DEPLOYMENT", "gpt-4o")
@@ -109,8 +103,29 @@ def classify_311_request(request_text: str) -> dict:
     #   response_format={"type": "json_object"}, temperature=0
     # TODO: Step 1.3 - Parse the JSON response with json.loads()
     raise NotImplementedError("Implement classify_311_request in Step 1")
+    """
+    client = _get_openai_client()
+
+    resp = client.chat.completions.create(
+        model=os.environ.get("AZURE_OPENAI_DEPLOYMENT", "gpt-4o"),
+        messages=[
+            {
+                "role": "system",
+                "content": (
+                    "You are an assistant for classifying Memphis 311 service requests. "
+                    "Classify the citizen's complaint into one of the following categories: "
+                    "Pothole, Noise Complaint, Trash/Litter, Street Light, Water/Sewer, Other. "
+                    "Respond with a JSON object with keys: category (one of the above), "
+                    "confidence (0-1), reasoning (brief explanation)."
+                ),
+            },
+            {"role": "user", "content": request_text},
+        ],
+        response_format={"type": "json_object"},
+        temperature=0,
+    )
 
-
+    return json.loads(resp.choices[0].message.content)
 # ---------------------------------------------------------------------------
 # TODO: Step 2 - Check content safety
 # ---------------------------------------------------------------------------
@@ -124,10 +139,34 @@ def check_content_safety(text: str) -> dict:
         dict with keys: safe (bool), categories (dict of category: severity)
     """
     # TODO: Step 2.1 - Get the Content Safety client
+    client = _get_content_safety_client()
+
     # TODO: Step 2.2 - Call client.analyze_text() with AnalyzeTextOptions
+    from azure.ai.contentsafety.models import AnalyzeTextOptions
+
+    result = client.analyze_text(AnalyzeTextOptions(text=text))
+    categories: dict[str, int] = {}
+
+    # Extract category severity scores from the result
+    if hasattr(result, "categories_analysis") and result.categories_analysis:
+        for category in result.categories_analysis:
+            # The category object has 'category' (the name) and 'severity' (the score)
+            cat_name = category.category
+            cat_severity = category.severity
+            categories[cat_name] = cat_severity
+    
+    # Determine safety: True if all categories have severity 0
+    is_safe = all(sev == 0 for sev in categories.values()) if categories else True
+    
     # TODO: Step 2.3 - Return safety results
-    raise NotImplementedError("Implement check_content_safety in Step 2")
-
+    return {
+        "safe": is_safe,
+        "categories": categories,
+    }
+    #raise NotImplementedError("Implement check_content_safety in Step 2")
+    
+    
+    
 
 # ---------------------------------------------------------------------------
 # TODO: Step 3 - Extract key phrases
@@ -142,9 +181,15 @@ def extract_key_phrases(text: str) -> list[str]:
         List of key phrase strings.
     """
     # TODO: Step 3.1 - Get the Language client
+    client = _get_language_client()
+
     # TODO: Step 3.2 - Call client.extract_key_phrases([text])
+    response = client.extract_key_phrases([text])
+    if response and not response[0].is_error:
+        return response[0].key_phrases
+    return []
     # TODO: Step 3.3 - Return the list of key phrases
-    raise NotImplementedError("Implement extract_key_phrases in Step 3")
+    #raise NotImplementedError("Implement extract_key_phrases in Step 3")
 
 
 def main():
@@ -186,7 +231,9 @@ def main():
     except NotImplementedError:
         print("Step 2 not implemented yet -- skipping content safety")
     except Exception as e:
-        print(f"Step 2 error: {e}")
+        print(f"Step 2 error: {type(e).__name__}: {e}")
+        import traceback
+        traceback.print_exc()
 
     # Step 3: Key phrase extraction
     key_phrases = None