Spaces:
Sleeping
Sleeping
File size: 3,200 Bytes
2f386cc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 |
import os
import about
import display
import download
import forms
import query
import repo
import streamlit as st
import utils
env_file_path = ".env"
log_file = "app.log"
temp_dir = "/tmp/chatgpt-code-review"
def app():
utils.load_environment_variables(env_file_path)
utils.set_environment_variables()
utils.configure_logging(log_file)
with utils.TempDirContext(temp_dir):
st.set_page_config(
page_title="ChatGPT Code Review",
)
session_state = st.session_state
st.title("ChatGPT Code Review :rocket:")
with st.expander("About ChatGPT Code Review"):
st.markdown(about.about_section, unsafe_allow_html=True)
st.write("")
default_repo_url = "https://github.com/domvwt/chatgpt-code-review"
repo_form = forms.RepoForm(default_repo_url)
with st.form("repo_url_form"):
repo_form.display_form()
repo_url, extensions = repo_form.get_form_data()
analyze_files_form = forms.AnalyzeFilesForm(session_state)
with st.form("analyze_files_form"):
if repo_form.clone_repo_button or session_state.get("code_files"):
if not session_state.get("code_files"):
session_state.code_files = (
repo.list_code_files_in_repository(
repo_url, extensions
)
)
analyze_files_form.display_form()
# Analyze the selected files
with st.spinner("Analyzing files..."):
if session_state.get("analyze_files"):
if session_state.get("selected_files"):
recommendations = query.analyze_code_files(
session_state.selected_files
)
# Display the recommendations
st.header("Recommendations")
first = True
recommendation_list = []
for rec in recommendations:
if not first:
st.write("---")
else:
first = False
st.subheader(display.escape_markdown(rec["code_file"]))
recommendation = (
rec["recommendation"] or "No recommendations"
)
st.markdown(recommendation)
with st.expander("View Code"):
extension = os.path.splitext(rec["code_file"])[1]
display.display_code(
rec["code_snippet"], extension
)
recommendation_list.append(rec)
if recommendation_list:
session_state.recommendation_list = recommendation_list
else:
st.error("Please select at least one file to analyze.")
st.stop()
st.write("")
download.download_markdown(session_state.get("recommendation_list"))
if __name__ == "__main__":
app()
|