Update app.py

app.py

@@ -1,10 +1,18 @@
-from fastapi import FastAPI
+
+
+
+from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 import requests
 import os
 from datetime import datetime, timedelta
 from groq import Groq
 from dotenv import load_dotenv
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
 
 # Load environment variables
 load_dotenv()
@@ -44,66 +52,120 @@ class RepositoryDetails(BaseModel):
     context: str
 
 def fetch_repository_changes(repo: str, days_back: int) -> list[str]:
-    since_date = (datetime.now() - timedelta(days=days_back)).isoformat()
-    headers = {
-        "Authorization": f"token {GITHUB_TOKEN}",
-        "Accept": "application/vnd.github.v3+json"
-    }
-    commits_url = f"{GITHUB_API_URL}/repos/{repo}/commits"
-    commits_params = {"since": since_date}
-    commits_response = requests.get(commits_url, headers=headers, params=commits_params)
-    commits = commits_response.json()
-    prs_url = f"{GITHUB_API_URL}/repos/{repo}/pulls"
-    prs_params = {"state": "all", "sort": "updated", "direction": "desc"}
-    prs_response = requests.get(prs_url, headers=headers, params=prs_params)
-    prs = prs_response.json()
-    changes = []
-    for commit in commits:
-        changes.append(f"Commit: {commit['commit']['message']}")
-    for pr in prs:
-        changes.append(f"PR: {pr['title']} - {pr['body']}")
-    return changes
+    """
+    Fetch recent commits and pull requests for a repository.
+    """
+    try:
+        logger.debug(f"Fetching changes for repository: {repo}")
+        since_date = (datetime.now() - timedelta(days=days_back)).isoformat()
+        headers = {
+            "Authorization": f"token {GITHUB_TOKEN}",
+            "Accept": "application/vnd.github.v3+json"
+        }
+
+        # Fetch commits
+        commits_url = f"{GITHUB_API_URL}/repos/{repo}/commits"
+        commits_params = {"since": since_date}
+        logger.debug(f"Fetching commits from: {commits_url}")
+        commits_response = requests.get(commits_url, headers=headers, params=commits_params)
+        commits_response.raise_for_status()  # Raise an exception for HTTP errors
+        commits = commits_response.json()
+        logger.debug(f"Found {len(commits)} commits for {repo}")
+
+        # Fetch pull requests
+        prs_url = f"{GITHUB_API_URL}/repos/{repo}/pulls"
+        prs_params = {"state": "all", "sort": "updated", "direction": "desc"}
+        logger.debug(f"Fetching pull requests from: {prs_url}")
+        prs_response = requests.get(prs_url, headers=headers, params=prs_params)
+        prs_response.raise_for_status()  # Raise an exception for HTTP errors
+        prs = prs_response.json()
+        logger.debug(f"Found {len(prs)} pull requests for {repo}")
+
+        # Extract changes
+        changes = []
+        for commit in commits:
+            changes.append(f"Commit: {commit['commit']['message']}")
+        for pr in prs:
+            changes.append(f"PR: {pr['title']} - {pr['body']}")
+
+        logger.debug(f"Total changes for {repo}: {len(changes)}")
+        return changes
+
+    except requests.exceptions.RequestException as e:
+        logger.error(f"Error fetching changes for {repo}: {e}")
+        raise HTTPException(status_code=500, detail=f"Error fetching changes for {repo}: {e}")
 
 def summarize_changes_with_deepseek(repo: str, changes: list[str]) -> dict:
-    prompt = f"""
-    The following changes were made to detection rules in the GitHub repository {repo}. 
-    Provide a detailed description of the changes and explain the context of why these changes are required:
-    {changes}
     """
-    response = groq_client.chat.completions.create(
-        model="deepseek-chat",
-        messages=[{"role": "user", "content": prompt}],
-        max_tokens=500,
-        temperature=0.7
-    )
-    summary = response.choices[0].message.content
-    description = summary.split("Description:")[1].split("Context:")[0].strip()
-    context = summary.split("Context:")[1].strip()
-    return {
-        "description": description,
-        "context": context
-    }
+    Use Groq's DeepSeek model to summarize changes and provide insights.
+    """
+    try:
+        logger.debug(f"Summarizing changes for repository: {repo}")
+        prompt = f"""
+        The following changes were made to detection rules in the GitHub repository {repo}. 
+        Provide a detailed description of the changes and explain the context of why these changes are required:
+        {changes}
+        """
+        logger.debug(f"Sending prompt to DeepSeek: {prompt}")
+        response = groq_client.chat.completions.create(
+            model="deepseek-chat",
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=500,
+            temperature=0.7
+        )
+        summary = response.choices[0].message.content
+        logger.debug(f"Received summary from DeepSeek: {summary}")
+
+        # Extract description and context from the summary
+        description = summary.split("Description:")[1].split("Context:")[0].strip()
+        context = summary.split("Context:")[1].strip()
+        logger.debug(f"Extracted description: {description}")
+        logger.debug(f"Extracted context: {context}")
+
+        return {
+            "description": description,
+            "context": context
+        }
+
+    except Exception as e:
+        logger.error(f"Error summarizing changes for {repo}: {e}")
+        raise HTTPException(status_code=500, detail=f"Error summarizing changes for {repo}: {e}")
 
 @app.get("/monitor", response_model=list[RepositoryDetails])
 async def monitor_repositories():
-    results = []
-    for repo in REPOSITORIES:
-        changes = fetch_repository_changes(repo, DAYS_BACK)
-        if changes:
-            summary = summarize_changes_with_deepseek(repo, changes)
-            results.append(RepositoryDetails(
-                repo_name=f"{repo} (+{len(changes)}, ✎{len(changes)})",
-                repo_url=f"https://github.com/{repo}",
-                changes="\n".join(changes),
-                description=summary["description"],
-                context=summary["context"]
-            ))
-        else:
-            results.append(RepositoryDetails(
-                repo_name=f"{repo} (No changes)",
-                repo_url=f"https://github.com/{repo}",
-                changes="No changes detected in the last 7 days.",
-                description="No changes detected.",
-                context="No context available."
-            ))
-    return results
+    """
+    Single API endpoint to fetch and summarize changes for all repositories.
+    """
+    try:
+        logger.debug("Starting to monitor repositories")
+        results = []
+        for repo in REPOSITORIES:
+            logger.debug(f"Processing repository: {repo}")
+            changes = fetch_repository_changes(repo, DAYS_BACK)
+
+            if changes:
+                logger.debug(f"Summarizing changes for {repo}")
+                summary = summarize_changes_with_deepseek(repo, changes)
+                results.append(RepositoryDetails(
+                    repo_name=f"{repo} (+{len(changes)}, ✎{len(changes)})",
+                    repo_url=f"https://github.com/{repo}",
+                    changes="\n".join(changes),
+                    description=summary["description"],
+                    context=summary["context"]
+                ))
+            else:
+                logger.debug(f"No changes detected for {repo}")
+                results.append(RepositoryDetails(
+                    repo_name=f"{repo} (No changes)",
+                    repo_url=f"https://github.com/{repo}",
+                    changes="No changes detected in the last 7 days.",
+                    description="No changes detected.",
+                    context="No context available."
+                ))
+
+        logger.debug("Finished monitoring repositories")
+        return results
+
+    except Exception as e:
+        logger.error(f"Error in monitor_repositories: {e}")
+        raise HTTPException(status_code=500, detail=str(e))