Added provision to accept the git repo url from UI

app.py

@@ -2,9 +2,12 @@
 
 # Import necessary modules from lib
 from lib.code_reviewer import CodeReviewer, ReviewManager
+import streamlit as st
 import os
 import glob
 
+# Streamlit UI for inputting GitHub repository details
+st.title("GitHub Repository Code Reviewer")
 # Variables for GitHub repository details
 REPO_OWNER = "vsagar100"
 REPO_NAME = "ansible_tf_provisioner"
@@ -15,11 +18,9 @@ GITHUB_TOKEN = "github_pat_11AF2YOZI0T6NzY3glKc04_40PRSN3Tl0dDhmrEdFZIbNMReQKktV
 #repo_url = f"https://api.github.com/repos/{REPO_OWNER}/{REPO_NAME}/zipball/main"
 GITHUB_REPO_URL = f"https://api.github.com/repos/{REPO_OWNER}/{REPO_NAME}/zipball/{GITHUB_BRANCH}"
 
-def main():
+if st.button("Review Code") and GITHUB_REPO_URL:
     # Directory structure setup
-    # Directory for storing downloaded GitHub repository
     download_directory = "downloaded_repo"
-    # Directory for storing output JSON reviews
     output_directory = "output_reviews"
 
     # Ensure the directories exist
@@ -30,20 +31,30 @@ def main():
     code_reviewer = CodeReviewer()
     review_manager = ReviewManager(reviewer=code_reviewer)
 
-    # Download GitHub repository
-    review_manager.download_repo(GITHUB_REPO_URL, GITHUB_TOKEN, download_directory)
+    try:
+        # Download GitHub repository
+        review_manager.download_repo(GITHUB_REPO_URL, GITHUB_TOKEN, download_directory)
 
-    # Find all YAML files in the downloaded repository
-    yaml_files = glob.glob(os.path.join(download_directory, "**", "*.yml"), recursive=True)
-    yaml_files += glob.glob(os.path.join(download_directory, "**", "*.yaml"), recursive=True)
+        # Find all YAML files in the downloaded repository
+        yaml_files = glob.glob(os.path.join(download_directory, "**", "*.yml"), recursive=True)
+        yaml_files += glob.glob(os.path.join(download_directory, "**", "*.yaml"), recursive=True)
 
-    # Process files and generate reviews
-    reviews = review_manager.process_files(yaml_files)
+        # Process files and generate reviews
+        reviews = review_manager.process_files(yaml_files)
 
-    # Save reviews to JSON
-    output_json_path = os.path.join(output_directory, "code_review_results.json")
-    review_manager.save_reviews_to_json(reviews, output_json_path)
-    print(f"Reviews saved to {output_json_path}")
+        # Save reviews to JSON
+        output_json_path = os.path.join(output_directory, "code_review_results.json")
+        review_manager.save_reviews_to_json(reviews, output_json_path)
 
-if __name__ == "__main__":
-    main()
+        # Display review results
+        st.success(f"Reviews saved to {output_json_path}")
+        for review in reviews:
+            st.subheader(f"Review for {review['filename']}")
+            st.text(review['review'])
+
+        # Provide download link for JSON results
+        with open(output_json_path, "r") as json_file:
+            st.download_button("Download JSON Results", json_file, file_name="code_review_results.json")
+
+    except Exception as e:
+        st.error(f"An error occurred: {str(e)}")

lib/code_reviewer.py

@@ -33,7 +33,7 @@ class CodeReviewer:
         Returns:
             Dict: The code standards in dictionary form.
         """
-        standards_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), "standards", "ansible_code_standards.json")
+        standards_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), "standards", "code_standards.json")
         with open(standards_path, 'r') as f:
             return json.load(f)
 
@@ -91,13 +91,13 @@ class ReviewManager:
             token (str): The GitHub personal access token for authentication.
             download_path (str): The path to extract the downloaded repository.
         """
-        headers = {"Authorization": f"token {token}"}
+        headers = {"Authorization": f"Bearer {token}"}
         response = requests.get(repo_url, headers=headers)
         if response.status_code == 200:
             with zipfile.ZipFile(io.BytesIO(response.content)) as zip_ref:
                 zip_ref.extractall(download_path)
         else:
-            raise Exception(f"Failed to download repository. Status code: {response.status_code}")
+            raise Exception(f"Failed to download repository. Status code: {response.status_code}, Message: {response.text}")
 
     def process_files(self, file_paths: List[str]) -> List[Dict[str, str]]:
         """