Update app.py

app.py

@@ -3,7 +3,7 @@ import requests
 import os
 import base64
 from urllib.parse import urlparse
-from groq import Groq
+import google.generativeai as genai
 
 
 # ── Helpers ────────────────────────────────────────────────────────────────────
@@ -35,11 +35,11 @@ def fetch_repo_files(owner: str, repo: str, github_token: str | None = None) ->
 
     tree = resp.json()
 
-    SKIP_DIRS   = {"node_modules", ".git", "__pycache__", "venv", "env",
-                   "dist", "build", ".next", "vendor", ".venv", "coverage"}
-    SKIP_EXTS   = {".png", ".jpg", ".jpeg", ".gif", ".svg", ".ico", ".pdf",
-                   ".zip", ".woff", ".ttf", ".eot", ".mp4", ".mp3", ".lock",
-                   ".bin", ".exe", ".so", ".dylib"}
+    SKIP_DIRS = {"node_modules", ".git", "__pycache__", "venv", "env",
+                 "dist", "build", ".next", "vendor", ".venv", "coverage"}
+    SKIP_EXTS = {".png", ".jpg", ".jpeg", ".gif", ".svg", ".ico", ".pdf",
+                 ".zip", ".woff", ".ttf", ".eot", ".mp4", ".mp3", ".lock",
+                 ".bin", ".exe", ".so", ".dylib"}
 
     candidates = []
     for item in tree.get("tree", []):
@@ -58,9 +58,9 @@ def fetch_repo_files(owner: str, repo: str, github_token: str | None = None) ->
     # Prioritise: README first, then root-level, then shallow paths
     def priority(p: str):
         name = p.lower()
-        if "readme" in name:        return 0
-        if p.count("/") == 0:       return 1
-        if p.count("/") == 1:       return 2
+        if "readme" in name:  return 0
+        if p.count("/") == 0: return 1
+        if p.count("/") == 1: return 2
         return 3 + p.count("/")
 
     selected = sorted(candidates, key=priority)[:18]
@@ -135,17 +135,17 @@ Order from most to least impactful.
 """
 
 
-def analyze_repo(repo_url: str, github_token: str, groq_api_key: str, progress=gr.Progress()):
+def analyze_repo(repo_url: str, github_token: str, gemini_api_key: str, progress=gr.Progress()):
     repo_url = repo_url or ""
     github_token = github_token or ""
-    groq_api_key = groq_api_key or ""
+    gemini_api_key = gemini_api_key or ""
 
     if not repo_url.strip():
         return "❌ **Error:** Please enter a GitHub repository URL."
 
-    api_key = os.environ.get("GROQ_API_KEY") or groq_api_key.strip()
+    api_key = os.environ.get("GEMINI_API_KEY") or gemini_api_key.strip()
     if not api_key:
-        return "❌ **Error:** Please enter your Groq API key (free at [console.groq.com](https://console.groq.com))."
+        return "❌ **Error:** Please enter your Gemini API key (free at [aistudio.google.com](https://aistudio.google.com))."
 
     try:
         progress(0.10, desc="Parsing repository URL…")
@@ -158,24 +158,29 @@ def analyze_repo(repo_url: str, github_token: str, groq_api_key: str, progress=g
         if not file_contents:
             return "❌ **Error:** No readable source files found. The repo may be empty or contain only binary files."
 
-        progress(0.65, desc="Running AI analysis with Llama 3 via Groq…")
-        client = Groq(api_key=api_key)
-        response = client.chat.completions.create(
-            model="llama-3.3-70b-versatile",
-            messages=[
-                {"role": "system", "content": SYSTEM_PROMPT},
-                {"role": "user",   "content": build_analysis_prompt(owner, repo, file_contents)},
-            ],
-            max_tokens=2_048,
-            temperature=0.3,
+        progress(0.65, desc="Running AI analysis with Gemini 2.0 Flash…")
+
+        # Configure Gemini
+        genai.configure(api_key=api_key)
+        model = genai.GenerativeModel(
+            model_name="gemini-2.0-flash",
+            system_instruction=SYSTEM_PROMPT
+        )
+
+        response = model.generate_content(
+            build_analysis_prompt(owner, repo, file_contents),
+            generation_config=genai.GenerationConfig(
+                max_output_tokens=2048,
+                temperature=0.3,
+            )
         )
 
         progress(1.0, desc="Done!")
 
-        report = response.choices[0].message.content
+        report = response.text
         header = (
             f"## 🔍 Analysis Report — `{owner}/{repo}`\n"
-            f"*{len(file_contents)} files sampled · Powered by Llama 3.3 70B via Groq*\n\n---\n\n"
+            f"*{len(file_contents)} files sampled · Powered by Gemini 2.0 Flash*\n\n---\n\n"
         )
         return header + report
 
@@ -200,7 +205,7 @@ with gr.Blocks(title="GitHub Repo Analyzer") as demo:
     gr.Markdown("# 🔍 GitHub Repo Analyzer", elem_id="title")
     gr.Markdown(
         "AI-powered **Code Quality & Documentation** analysis — paste any public repo and get a full report in seconds.\n\n"
-        "_Powered by **Llama 3.3 70B** via Groq — blazing fast & free._",
+        "_Powered by **Gemini 2.0 Flash** — blazing fast & free._",
         elem_id="sub",
     )
 
@@ -216,13 +221,13 @@ with gr.Blocks(title="GitHub Repo Analyzer") as demo:
 
     with gr.Accordion("⚙️ API Keys", open=False):
         gr.Markdown(
-            "💡 _If the Space owner has set `GROQ_API_KEY` as a HF Secret, you don't need to fill this in._\n\n"
-            "Get a **free** Groq API key at [console.groq.com](https://console.groq.com) — no credit card needed."
+            "💡 _If the Space owner has set `GEMINI_API_KEY` as a HF Secret, you don't need to fill this in._\n\n"
+            "Get a **free** Gemini API key at [aistudio.google.com](https://aistudio.google.com) — no credit card needed."
         )
         with gr.Row():
-            groq_key_input = gr.Textbox(
-                label="Groq API Key (free)",
-                placeholder="gsk_xxxxxxxxxxxx",
+            gemini_key_input = gr.Textbox(
+                label="Gemini API Key (free)",
+                placeholder="AIza_xxxxxxxxxxxx",
                 type="password",
                 lines=1,
             )
@@ -237,7 +242,7 @@ with gr.Blocks(title="GitHub Repo Analyzer") as demo:
 
     analyze_btn.click(
         fn=analyze_repo,
-        inputs=[repo_url_input, github_token_input, groq_key_input],
+        inputs=[repo_url_input, github_token_input, gemini_key_input],
         outputs=output_md,
     )
 
@@ -252,7 +257,7 @@ with gr.Blocks(title="GitHub Repo Analyzer") as demo:
     )
 
     gr.Markdown(
-        "---\nBuilt with **Llama 3.3 70B via Groq** · [Get your free key](https://console.groq.com) · "
+        "---\nBuilt with **Gemini 2.0 Flash** · [Get your free key](https://aistudio.google.com) · "
         "Made by [Worply](https://worply.tech)",
         elem_id="footer",
     )