Update app.py

app.py

@@ -2,20 +2,39 @@ import gradio as gr
 from datetime import datetime, timedelta
 import google.generativeai as genai
 from github import Github, GithubException
+import gitlab
 import docx
 import tempfile
+import requests
 
-def generate_release_notes(github_repo, github_token, gemini_api_key, start_date, end_date):
+def generate_release_notes(git_provider, repo_url, personal_access_token, gemini_api_key, start_date, end_date):
     try:
-        g = Github(github_token)
-        
-        repo = g.get_repo(github_repo)
-        
         start_date = datetime.strptime(start_date, "%Y-%m-%d")
         end_date = datetime.strptime(end_date, "%Y-%m-%d")
-        commits = list(repo.get_commits(since=start_date, until=end_date))
-        
-        commit_messages = [commit.commit.message for commit in commits]
+
+        if git_provider == "GitHub":
+            g = Github(personal_access_token)
+            repo = g.get_repo(repo_url)
+            commits = list(repo.get_commits(since=start_date, until=end_date))
+            commit_messages = [commit.commit.message for commit in commits]
+        elif git_provider == "GitLab":
+            gl = gitlab.Gitlab(url='https://gitlab.com', private_token=personal_access_token)
+            project = gl.projects.get(repo_url)
+            commits = project.commits.list(since=start_date.isoformat(), until=end_date.isoformat())
+            commit_messages = [commit.message for commit in commits]
+        elif git_provider == "Gitea":
+            base_url = "https://gitea.com/api/v1"
+            headers = {"Authorization": f"token {personal_access_token}"}
+            response = requests.get(f"{base_url}/repos/{repo_url}/commits", headers=headers, params={
+                "since": start_date.isoformat(),
+                "until": end_date.isoformat()
+            })
+            response.raise_for_status()
+            commits = response.json()
+            commit_messages = [commit['commit']['message'] for commit in commits]
+        else:
+            return "Unsupported Git provider", None, None
+
         commit_text = "\n".join(commit_messages)
         
         if not commit_text:
@@ -81,13 +100,6 @@ def generate_release_notes(github_repo, github_token, gemini_api_key, start_date
         
         return release_notes, docx_path, md_path
     
-    except GithubException as e:
-        if e.status == 401:
-            return "Error: Invalid GitHub token or insufficient permissions.", None, None
-        elif e.status == 404:
-            return f"Error: Repository not found. Please check the GitHub repository name. Attempted to access: {github_repo}", None, None
-        else:
-            return f"GitHub API error: {str(e)}", None, None
     except Exception as e:
         return f"An error occurred: {str(e)}", None, None
 
@@ -97,8 +109,12 @@ default_start_date = default_end_date - timedelta(days=30)
 iface = gr.Interface(
     fn=generate_release_notes,
     inputs=[
-        gr.Textbox(label="GitHub Repository Name", placeholder="MicroHealthLLC/maiko-assistant"),
-        gr.Textbox(label="GitHub Personal Access Token", type="password"),
+        gr.Dropdown(
+            choices=["GitHub", "GitLab", "Gitea"],
+            label="Git Provider"
+        ),
+        gr.Textbox(label="Repository URL", placeholder="owner/repo"),
+        gr.Textbox(label="Personal Access Token", type="password"),
         gr.Textbox(label="Gemini API Key", type="password"),
         gr.Textbox(
             label="Start Date",
@@ -117,7 +133,7 @@ iface = gr.Interface(
         gr.File(label="Download Release Notes (Markdown)")
     ],
     title="Automated Release Notes Generator",
-    description="Generate release notes based on GitHub commits using Gemini AI. Enter start and end dates (YYYY-MM-DD) to define the time range for commits.",
+    description="Generate release notes based on Git commits using Gemini AI. Select a Git provider, enter repository details, and specify the date range for commits.",
     allow_flagging="never",
     theme="default",
     analytics_enabled=False,