Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from github import Github, GithubException
 import os
 import requests
+import re
 
 # Załaduj token z pliku .env lub ustaw bezpośrednio
 GITHUB_TOKEN = os.getenv('GITHUB_TOKEN')
@@ -15,6 +16,14 @@ def get_file_content(owner, repo_name, path, branch="main"):
     else:
         return f"Błąd pobierania pliku: {response.status_code}"
 
+def extract_repo_info(url):
+    """Wyodrębnia nazwę użytkownika i repozytorium z linku GitHub."""
+    match = re.search(r"github\.com/([^/]+)/([^/]+)", url)
+    if match:
+        return match.group(1), match.group(2)
+    else:
+        return None, None
+
 
 def github_tool(
     action: str,
@@ -33,7 +42,8 @@ def github_tool(
     labels: str = None,  # etykiety oddzielone przecinkami
     tag: str = None,
     name: str = None,  # nazwa release
-    file_url: str = None, # URL pliku do pobrania
+    file_url: str = None, # URL pliku do pobrania,
+    repo_url: str = None,  # Link do repozytorium
 ):
     """Narzędzie do zarządzania repozytoriami GitHub."""
     user = g.get_user()
@@ -251,10 +261,40 @@ def github_tool(
             }
             return info  # Zwraca słownik z informacjami
             
-        elif action == "get_file_content":  # Dodana akcja
+        elif action == "get_file_content":
             if not all([owner, repo_name, path]):
                 raise ValueError("Brakujące parametry: owner, repo_name, path")
-            return get_file_content(owner, repo_name, path, branch)  
+            return get_file_content(owner, repo_name, path, branch) 
+        
+        elif action == "analyze_repository_by_url":
+            if not repo_url:
+                raise ValueError("Brakujący parametr: repo_url")
+
+            owner, repo_name = extract_repo_info(repo_url)
+            if not owner or not repo_name:
+                raise ValueError("Nieprawidłowy link do repozytorium")
+
+            try:
+                repo = g.get_repo(f"{owner}/{repo_name}")
+
+                # Pobierz listę plików i katalogów
+                contents = repo.get_contents("")
+
+                # Iteruj po liście i pobieraj zawartość plików
+                file_analyses = []
+                for content in contents:
+                    if content.type == "file":
+                        file_content = content.decoded_content.decode()
+                        file_analyses.append({
+                            "name": content.name,
+                            "path": content.path,
+                            "content": file_content,
+                            # Możesz dodać tutaj analizę zawartości pliku
+                        })
+                return file_analyses
+
+            except GithubException as e:
+                return f"Błąd GitHub: {str(e)}"
 
         else:
             raise ValueError(f"Nieznana akcja: {action}")
@@ -292,7 +332,8 @@ with gr.Blocks() as demo:
                 "list_forks",
                 "list_files",
                 "get_repository_info", 
-                "get_file_content",  # Dodana akcja
+                "get_file_content",
+                "analyze_repository_by_url",
             ],
             label="Akcja",
         )
@@ -312,6 +353,7 @@ with gr.Blocks() as demo:
         tag = gr.Textbox(label="Tag")
         release_name = gr.Textbox(label="Nazwa release")  # Zmieniona nazwa
         file_url = gr.Textbox(label="URL pliku") # Dodane pole
+        repo_url = gr.Textbox(label="Link do repozytorium") # Dodane pole
     
     with gr.Row():
         run_button = gr.Button("Wykonaj")
@@ -337,6 +379,7 @@ with gr.Blocks() as demo:
             tag,
             release_name,  # Użycie zmienionej nazwy
             file_url, # Dodany argument
+            repo_url, # Dodany argument
         ],
         outputs=output,
         api_name="github_tool"