test/bad-code

.github/workflows/ci.yml

@@ -0,0 +1,26 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.11'
+
+      - name: Install dependencies
+        run: pip install -r requirements.txt
+
+      - name: Run tests
+        run: pytest -v

.github/workflows/pr-review.yml

@@ -0,0 +1,56 @@
+name: AI Code Review
+
+on:
+  pull_request:
+    branches: [main]
+
+permissions:
+  contents: read
+  pull-requests: write
+
+jobs:
+  review:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.11'
+
+      - name: Install dependencies
+        run: pip install -r requirements.txt
+
+      - name: Get PR diff
+        run: |
+          git diff origin/main...HEAD > pr_diff.txt
+
+      - name: Run AI review
+        id: ai-review
+        env:
+          GEMINI_API_KEY: ${{ secrets.GEMINI_API_KEY }}
+        run: |
+          REVIEW=$(python scripts/ai_review.py pr_diff.txt)
+          echo "review<<EOF" >> $GITHUB_OUTPUT
+          echo "$REVIEW" >> $GITHUB_OUTPUT
+          echo "EOF" >> $GITHUB_OUTPUT
+
+      - name: Post review comment
+        uses: actions/github-script@v7
+        env:
+          REVIEW: ${{ steps.ai-review.outputs.review }}
+        with:
+          script: |
+            const review = process.env.REVIEW;
+
+            await github.rest.issues.createComment({
+              owner: context.repo.owner,
+              repo: context.repo.repo,
+              issue_number: context.issue.number,
+              body: `## 🤖 AI Code Review\n\n${review}\n\n---\n*Powered by Gemini AI*`
+            });

.gitignore

@@ -31,3 +31,5 @@ Thumbs.db
 
 # Ruff
 .ruff_cache/
+
+.ai_review.py

app.py

@@ -14,3 +14,9 @@ def is_even(n: int) -> bool:
 def reverse_string(s: str) -> str:
     """Reverse a string."""
     return s[::-1]
+
+
+def multiply(a: int, b: int) -> int:
+    """Multiply two numbers together."""
+    return a * b
+

requirements.txt

@@ -1,5 +1,5 @@
-# Production dependencies (none for this simple app)
-
+# Production dependencies 
+google-genai>=1.0.0
 # Development/testing dependencies
 pytest>=7.0.0
 build>=1.0.0

scripts/ai_review.py

@@ -0,0 +1,44 @@
+from google import genai
+import sys
+
+client = genai.Client()
+
+def review_code(diff_text):
+    """Send a code diff to Gemini for review."""
+    prompt = f"""You are an expert code reviewer. Review the following code diff and provide feedback.
+
+Focus on:
+1. Security vulnerabilities
+2. Bug risks
+3. Performance issues
+4. Best practice violations
+
+For each issue found, provide:
+- Severity: HIGH / MEDIUM / LOW
+- Description of the issue
+- Suggested fix
+
+If the code looks good, say so.
+
+Code diff to review:
+
+{diff_text}
+
+
+Provide your review in a clear, structured format."""
+
+    response = client.models.generate_content(
+        model="gemini-2.5-flash", contents=prompt
+    )
+    return response.text
+
+if __name__ == "__main__":
+    if len(sys.argv) > 1:
+        diff_file = sys.argv[1]
+        with open(diff_file, "r") as f:
+            diff_content = f.read()
+    else:
+        diff_content = sys.stdin.read()
+
+    review = review_code(diff_content)
+    print(review)

scripts/sample_diff.txt

@@ -0,0 +1,17 @@
+diff --git a/app.py b/app.py
+index 1234567..abcdefg 100644
+--- a/app.py
++++ b/app.py
+@@ -1,5 +1,12 @@
+ """Simple utility functions"""
+
++import sqlite3
++
++def get_user(username):
++    conn = sqlite3.connect("users.db")
++    query = f"SELECT * FROM users WHERE name = '{username}'"
++    return conn.execute(query).fetchone()
++
+ def add(a: int, b: int) -> int:
+     """Add two numbers together."""
+     return a + b

tests/test_app.py

@@ -1,6 +1,6 @@
 """Tests for app.py - you'll add more!"""
 
-from app import add, is_even, reverse_string
+from app import add, is_even, reverse_string, multiply
 
 
 class TestMath:
@@ -22,3 +22,11 @@ def test_reverse(self):
     def test_is_even(self):
         assert is_even(4) is True
         assert is_even(3) is False
+
+
+class TestAdd:
+    def test_add_positive(self):
+        assert add(2, 3) == 5
+
+
+