Update app.py

app.py

@@ -1,6 +1,3 @@
-
-
-
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 import requests
@@ -17,7 +14,7 @@ logger = logging.getLogger(__name__)
 # Load environment variables
 load_dotenv()
 
-# Configuration
+# Configuration (Use environment variables instead of hardcoding)
 GITHUB_TOKEN = "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"
 GROQ_API_KEY = "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"
 REPOSITORIES = [
@@ -66,18 +63,18 @@ def fetch_repository_changes(repo: str, days_back: int) -> list[str]:
         # Fetch commits
         commits_url = f"{GITHUB_API_URL}/repos/{repo}/commits"
         commits_params = {"since": since_date}
-        logger.debug(f"Fetching commits from: {commits_url}")
+        logger.debug(f"Fetching commits from: {commits_url} with params: {commits_params}")
         commits_response = requests.get(commits_url, headers=headers, params=commits_params)
-        commits_response.raise_for_status()  # Raise an exception for HTTP errors
+        commits_response.raise_for_status()
         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}")
+        logger.debug(f"Fetching pull requests from: {prs_url} with params: {prs_params}")
         prs_response = requests.get(prs_url, headers=headers, params=prs_params)
-        prs_response.raise_for_status()  # Raise an exception for HTTP errors
+        prs_response.raise_for_status()
         prs = prs_response.json()
         logger.debug(f"Found {len(prs)} pull requests for {repo}")
 
@@ -86,7 +83,9 @@ def fetch_repository_changes(repo: str, days_back: int) -> list[str]:
         for commit in commits:
             changes.append(f"Commit: {commit['commit']['message']}")
         for pr in prs:
-            changes.append(f"PR: {pr['title']} - {pr['body']}")
+            updated_at = datetime.strptime(pr["updated_at"], "%Y-%m-%dT%H:%M:%SZ")
+            if updated_at >= datetime.now() - timedelta(days=days_back):
+                changes.append(f"PR: {pr['title']} - {pr['body'] or 'No description'}")
 
         logger.debug(f"Total changes for {repo}: {len(changes)}")
         return changes
@@ -102,11 +101,14 @@ def summarize_changes_with_deepseek(repo: str, changes: list[str]) -> dict:
     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}
+        Analyze the following changes made to detection rules in the GitHub repository {repo}:
+        {', '.join(changes)}
+        
+        Provide a detailed response with two sections:
+        - Description: Summarize what changes were made.
+        - Context: Explain why these changes might be required.
         """
-        logger.debug(f"Sending prompt to DeepSeek: {prompt}")
+        logger.debug(f"Sending prompt to DeepSeek: {prompt[:100]}...")  # Truncate for brevity in logs
         response = groq_client.chat.completions.create(
             model="deepseek-chat",
             messages=[{"role": "user", "content": prompt}],
@@ -114,18 +116,18 @@ def summarize_changes_with_deepseek(repo: str, changes: list[str]) -> dict:
             temperature=0.7
         )
         summary = response.choices[0].message.content
-        logger.debug(f"Received summary from DeepSeek: {summary}")
+        logger.debug(f"Received summary from DeepSeek: {summary[:100]}...")
 
-        # 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}")
+        # Extract description and context with fallback
+        description = "Description not found."
+        context = "Context not found."
+        if "Description:" in summary and "Context:" in summary:
+            description = summary.split("Description:")[1].split("Context:")[0].strip()
+            context = summary.split("Context:")[1].strip()
+        else:
+            description = summary  # Fallback to full summary if sections aren't clear
 
-        return {
-            "description": description,
-            "context": context
-        }
+        return {"description": description, "context": context}
 
     except Exception as e:
         logger.error(f"Error summarizing changes for {repo}: {e}")
@@ -147,7 +149,7 @@ async def monitor_repositories():
                 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_name=f"{repo} (+{len(changes)})",
                     repo_url=f"https://github.com/{repo}",
                     changes="\n".join(changes),
                     description=summary["description"],
@@ -168,4 +170,8 @@ async def monitor_repositories():
 
     except Exception as e:
         logger.error(f"Error in monitor_repositories: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)