Update app.py

app.py

@@ -6,24 +6,6 @@ import requests
 # Hugging Face Inference Client
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-# GitHub API details
-GITHUB_USERNAME = "YOUR_USERNAME"
-GITHUB_REPOSITORY = "YOUR_REPOSITORY"
-GITHUB_API_TOKEN = os.getenv("GITHUB_API_TOKEN")
-
-# Function to fetch GitHub issues
-def fetch_github_issues():
-  url = f"https://api.github.com/repos/{GITHUB_USERNAME}/{GITHUB_REPOSITORY}/issues"
-  headers = {
-    "Authorization": f"Bearer {GITHUB_API_TOKEN}",
-    "Accept": "application/vnd.github.v3+json"
-  }
-  response = requests.get(url, headers=headers)
-  if response.status_code == 200:
-    return response.json()
-  else:
-    raise Exception(f"Error fetching issues: {response.status_code}")
-
 # Function to analyze issues and provide solutions
 def analyze_issues(issue_text, model_name):
   nlp = pipeline("text-generation", model=model_name)
@@ -38,7 +20,13 @@ def respond(
     max_tokens,
     temperature,
     top_p,
+    github_api_token,
+    github_username,
+    github_repository
 ):
+    global GITHUB_API_TOKEN
+    GITHUB_API_TOKEN = github_api_token
+
     messages = [{"role": "system", "content": system_message}]
 
     for val in history:
@@ -48,22 +36,46 @@ def respond(
             messages.append({"role": "assistant", "content": val[1]})
 
     if message.startswith("/github"):
-        try:
-            issues = fetch_github_issues()
-            issue_list = "\n".join([f"{i+1}. {issue['title']}" for i, issue in enumerate(issues)])
-            yield f"Available GitHub Issues:\n{issue_list}\n\nEnter the issue number to analyze:"
-        except Exception as e:
-            yield f"Error fetching GitHub issues: {e}"
+        if not GITHUB_API_TOKEN:
+            yield "Please enter your GitHub API token first. [Click here to get your token](https://github.com/settings/tokens)"
+        else:
+            try:
+                url = f"https://api.github.com/repos/{github_username}/{github_repository}/issues"
+                headers = {
+                    "Authorization": f"Bearer {GITHUB_API_TOKEN}",
+                    "Accept": "application/vnd.github.v3+json"
+                }
+                response = requests.get(url, headers=headers)
+                if response.status_code == 200:
+                    issues = response.json()
+                    issue_list = "\n".join([f"{i+1}. {issue['title']}" for i, issue in enumerate(issues)])
+                    yield f"Available GitHub Issues:\n{issue_list}\n\nEnter the issue number to analyze:"
+                else:
+                    yield f"Error fetching GitHub issues: {response.status_code}"
+            except Exception as e:
+                yield f"Error fetching GitHub issues: {e}"
     elif message.isdigit():
-        try:
-            issue_number = int(message) - 1
-            issues = fetch_github_issues()
-            issue = issues[issue_number]
-            issue_text = issue['title'] + "\n\n" + issue['body']
-            resolution = analyze_issues(issue_text, "gpt2") # Default to gpt2 for now
-            yield f"Resolution for Issue '{issue['title']}':\n{resolution}"
-        except Exception as e:
-            yield f"Error analyzing issue: {e}"
+        if not GITHUB_API_TOKEN:
+            yield "Please enter your GitHub API token first. [Click here to get your token](https://github.com/settings/tokens)"
+        else:
+            try:
+                issue_number = int(message) - 1
+                url = f"https://api.github.com/repos/{github_username}/{github_repository}/issues"
+                headers = {
+                    "Authorization": f"Bearer {GITHUB_API_TOKEN}",
+                    "Accept": "application/vnd.github.v3+json"
+                }
+                response = requests.get(url, headers=headers)
+                if response.status_code == 200:
+                    issues = response.json()
+                    issue = issues[issue_number]
+                    issue_text = issue['title'] + "\n\n" + issue['body']
+                    resolution = analyze_issues(issue_text, "gpt2")  # Default to gpt2 for now
+                    yield f"Resolution for Issue '{issue['title']}':\n{resolution}"
+                else:
+                    yield f"Error fetching GitHub issues: {response.status_code}"
+            except Exception as e:
+                yield f"Error analyzing issue: {e}"
     else:
         messages.append({"role": "user", "content": message})
 
@@ -80,6 +92,11 @@ def respond(
             yield response
 
 with gr.Blocks() as demo:
+    with gr.Row():
+        github_api_token = gr.Textbox(label="GitHub API Token", type="password")
+        github_username = gr.Textbox(label="GitHub Username")
+        github_repository = gr.Textbox(label="GitHub Repository")
+
     chatbot = gr.ChatInterface(
         respond,
         additional_inputs=[
@@ -93,6 +110,9 @@ with gr.Blocks() as demo:
                 step=0.05,
                 label="Top-p (nucleus sampling)",
             ),
+            github_api_token,
+            github_username,
+            github_repository
         ],
     )