FIRST COMMIT LESGO EZCLAP

.gitignore

@@ -0,0 +1,3 @@
+.env
+venv/
+__pycache__

Dockerfile

@@ -0,0 +1,16 @@
+FROM python:3
+
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+
+COPY --chown=user . $HOME/github_bot
+
+WORKDIR $HOME/github_bot
+
+RUN mkdir $HOME/.cache
+
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+CMD ["gunicorn", "-w", "5", "-b", "0.0.0.0:7860","app:app"]

jwt_setup.py

@@ -0,0 +1,69 @@
+from datetime import datetime
+import jwt
+import requests
+import base64
+
+
+def decode_string(base64_string: str) -> str:
+  """Decodes a Base64 string.
+  
+  NOTE:
+  This workaround was required because .env cannot handle multi-line strings, so it was required to convert it to a single line. There are mutliple ways to achieve the same result.
+  """
+
+  base64_bytes = base64_string.encode('utf-8')
+  string_bytes = base64.b64decode(base64_bytes)
+  string = string_bytes.decode('utf-8')
+  return string
+
+
+def get_installation_access_token(
+    signing_key: str, app_id: str, installation_id: str
+) -> str:
+    """
+    Obtain and return a GitHub installation access token.
+
+    Arguments:
+        signing_key: base64 encoded signing key (you'll get this from the .env)
+        app_id: The application ID
+        installation_id: The ID of the app installation.
+
+    Returns:
+        The installation access token obtained from GitHub.
+    """
+
+    # Refer https://github.com/orgs/community/discussions/48186
+    # Refer https://stackoverflow.com/questions/77325437/how-do-i-get-an-github-app-installation-token-to-authenticate-cloning-a-reposito
+    now = int(datetime.now().timestamp())
+
+    signing_key = decode_string(signing_key)
+    signing_key = jwt.jwk_from_pem(signing_key.encode())
+    payload = {"iat": now, "exp": now + 600, "iss": app_id}
+    jwt_instance = jwt.JWT()
+    encoded_jwt = jwt_instance.encode(payload, signing_key, alg="RS256")
+
+    response = requests.post(
+        "https://api.github.com/app/installations/" f"{installation_id}/access_tokens",
+        headers={
+            "Authorization": f"Bearer {encoded_jwt}",
+            "Accept": "application/vnd.github+json",
+            "X-GitHub-Api-Version": "2022-11-28",
+        },
+    )
+    if not 200 <= response.status_code < 300:
+        raise RuntimeError(
+            "Unable to get token. Status code was "
+            f"{response.status_code}, body was {response.text}."
+        )
+
+    return response.json()["token"]
+
+
+if __name__ == "__main__":
+    import os
+
+    print(get_installation_access_token(
+        os.environ.get('CHATBOT_PUB_KEY'),
+        os.environ.get('APP_ID'),
+        os.environ.get('INSTALLATION_ID')    
+    ))

server.py

@@ -0,0 +1,49 @@
+from flask import Flask, request
+import requests
+import json
+import os
+
+from jwt_setup import get_installation_access_token
+
+
+app = Flask(__name__)
+
+def get_llm_output(query: str):
+    LLM_output = requests.post('https://robinroy03-fury-engine.hf.space/api/google/generate', json={"query": query, "llm": "gemini-1.5-pro", "knn": 3, "stream": False})
+    LLM_output = json.loads(LLM_output.text)
+    LLM_output = LLM_output['response'] + '\n\n**References**\n' + LLM_output['references']
+
+    return LLM_output
+
+
+def post_discussion_comment(discussion_id: str, body: str):
+    token = get_installation_access_token(
+        os.environ.get('CHATBOT_PUB_KEY'),
+        os.environ.get('APP_ID'),
+        os.environ.get('INSTALLATION_ID')
+    )
+
+    headers = {"Authorization": f"token {token}"}
+    query = "mutation AddDiscussionComment($discussionId: ID!, $body: String!) {addDiscussionComment(input: {discussionId: $discussionId, body: $body}) {clientMutationId}}"
+    variables = {"discussionId": discussion_id, "body": body}
+
+    requests.post('https://api.github.com/graphql', json={'query': query, 'variables': variables}, headers=headers)
+
+
+@app.route("/", methods=['POST'])
+def main():
+    payload = request.get_json()
+    URL = "https://api.github.com/graphql"
+
+    if 'discussion' in payload and 'comment' not in payload:
+        discussion_id = payload['discussion']['node_id']
+        discussion_title = payload['discussion']['title']
+        discussion_body = payload['discussion']['body']
+        print(discussion_id)
+
+        post_discussion_comment(discussion_id=discussion_id, body=get_llm_output(f"{discussion_title}\n{discussion_body}"))
+
+        return {"status": True}
+    
+    else:
+        return {"status": False}