Update app.py

app.py

@@ -1,75 +1,70 @@
-import os
 import streamlit as st
 from github import Github
-from github import GithubException
-from altair import Chart
+from github import UnknownObjectException
+import logging
 
-# Set up the Streamlit app
-st.set_page_config(page_title="GitHub File Search", page_icon=":mag_right:", theme="dark")
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
-# Retrieve the GitHub API token from the environment variable
-access_token = os.environ["GITHUB_TOKEN"]
+st.set_page_config(page_title="GitHub File Search", page_icon=":mag_right:")
+access_token = "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"
 
-# Define a function to get all the repositories that contain a file with the given name
-def get_repositories_with_file(filename):
-    # Create a GitHub client
-    g = Github(access_token)
 
-    # Search for the file on GitHub
-    query = f"filename:{filename}"
-    repositories = g.search_repositories(query=query, sort="stars", order="desc")
+def find_file(repo, path, filename):
+    contents = repo.get_contents(path)
+    for content in contents:
+        if content.type == "dir":
+            result = find_file(repo, content.path, filename)
+            if result:
+                return result
+        elif content.name == filename:
+            return content
+    return None
+
 
-    # Initialize a list to store the repository information
+def get_repositories_with_file(filename):
+    g = Github(access_token)
+    repositories = g.search_repositories(query=f"filename:{filename}", sort="stars", order="desc")
     repo_info = []
 
-    # Loop through the repositories
     for repo in repositories:
-        # Get the contents of the repository
-        contents = repo.get_contents("")
-
-        # Check if the repository contains a file with the given name
-        for content in contents:
-            if content.name == filename:
+        try:
+            content = find_file(repo, "", filename)
+            if content:
                 repo_info.append({
                     "name": repo.name,
                     "description": repo.description,
                     "url": repo.html_url
                 })
-                break
-
-    # Return the repository information
+        except UnknownObjectException as e:
+            logger.warning(f"File '{filename}' not found in repository '{repo.full_name}'.")
+            st.warning(f"File '{filename}' not found in repository '{repo.full_name}'.")
     return repo_info
 
-# Define the Streamlit app
+
 def app():
-    # Set up the user interface
     st.title("GitHub File Search")
     st.write("Enter a file name to search for on GitHub:")
     filename = st.text_input("File name")
-
     if filename:
-        # Get the repositories with the file
         repo_info = get_repositories_with_file(filename)
-
-        # Display the repository information
         if len(repo_info) > 0:
             st.success(f"Found {len(repo_info)} repositories with the file '{filename}':")
             for repo in repo_info:
                 st.write(f"- **{repo['name']}**: {repo['description']}")
                 st.write(f"  URL: [{repo['url']}]({repo['url']})")
         else:
+            logger.warning(f"No repositories found with the file '{filename}'.")
             st.warning(f"No repositories found with the file '{filename}'.")
 
-        # Create a chart to show the number of repositories found for each file name
-        chart = Chart(data=repo_info).mark_bar().encode(
-            x='name',
-            y='count',
-            color='name'
-        )
+    st.sidebar.title("GitHub File Search Options")
+    show_logs = st.sidebar.checkbox("Show Logs")
+    if show_logs:
+        st.sidebar.subheader("Logs")
+        with open("app.log", "r") as log_file:
+            logs = log_file.read()
+        st.sidebar.text(logs)
 
-        # Display the chart
-        st.plotly_chart(chart)
 
-# Run the Streamlit app
 if __name__ == "__main__":
-    app()
+    app()