Update gemini_review.py

Author: richabanker

hack/gemini_review.py

@@ -1,19 +1,14 @@
 import google.generativeai as genai
 import os
 from github import Github
-import base64
 
-def get_pr_diff():
-    """Retrieves, base64 decodes, and cleans the PR diff from environment variables."""
-    encoded_diff = os.environ.get('INPUT_DIFF') or ""
-    try:
-        decoded_diff = base64.b64decode(encoded_diff).decode('utf-8')
-    except Exception as e:
-        print(f"Error decoding base64: {e}")
-        return "" #Return empty string on error.
-
-    cleaned_diff = decoded_diff.replace("DIFF<<EOF", "").replace("EOF", "").strip()
-    return cleaned_diff
+def get_pr_diff(repo_name, pr_number, github_token):
+    """Retrieves and cleans the PR diff using PyGithub."""
+    g = Github(github_token)
+    repo = g.get_repo(repo_name)
+    pr = repo.get_pull(pr_number)
+    diff = pr.get_commits().files.patch  # Get diff from the latest commit
+    return diff
 
 def generate_gemini_review(diff, api_key):
     """Generates a code review using the Gemini API."""
@@ -45,7 +40,8 @@ def main():
     pr_number = int(os.environ.get('PR_NUMBER'))
     repo_name = os.environ.get('GITHUB_REPOSITORY')
     github_token = os.environ.get('GITHUB_TOKEN')
-    diff = get_pr_diff()
+
+    diff = get_pr_diff(repo_name, pr_number, github_token)  # Get diff directly
 
     review_comment = generate_gemini_review(diff, api_key)