Spaces:
Sleeping
Sleeping
File size: 5,487 Bytes
c78d99c 98c2214 c78d99c |
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 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 |
import streamlit as st
from openai_client import (
get_code_review_response,
refactor_code,
code_feedback,
suggest_best_practices,
remove_code_errors,
)
def main():
st.title("CodeMentor - (AI-Enhanced Code Collaboration Tool)")
st.subheader("Collaborate, Refactor, and Optimize with AI.")
st.write(
"A smart tool for distributed teams to automate code reviews, refactor efficiently, and get real-time AI-driven feedback."
)
# Instructions
st.write(
"Upload a file or paste your code below to get an AI-generated code review."
)
# Input Methods: File Upload or Text Area
uploaded_file = st.file_uploader(
"Upload a code file (Max 500 lines)", type=["py", "js", "txt"]
)
code_input = st.text_area("Or paste your code here (Max 1000 words)", height=300)
# Limit input size for code
if uploaded_file:
code = uploaded_file.read().decode("utf-8")
if len(code.splitlines()) > 500:
st.error(
"File is too large! Please upload a file with a maximum of 500 lines."
)
code = None # Reset code if it's too large
else:
st.success(f"File uploaded: {uploaded_file.name}")
elif code_input:
code = code_input
if len(code.split()) > 1000:
st.error("Code exceeds 1000 words! Please shorten your code.")
code = None # Reset code if it's too large
else:
code = None
# Button to trigger code review
if st.button("Get Code Review") and code:
with st.spinner("Processing..."):
# Call the OpenAI API to get code review
review = get_code_review_response(code)
st.subheader("Code Review Results:")
st.write(review)
# Provide download option
st.download_button(
label="Download Code Review",
data=review,
file_name="code_review.txt",
mime="text/plain",
)
st.success("You can download the code review as code_review.txt")
# Button to trigger code refactoring
if st.button("Refactor Code") and code:
with st.spinner("Refactoring your code..."):
refactored_code = refactor_code(code)
st.subheader("Refactored Code:")
st.write(st.code(refactored_code, language='python'))
# Provide download option for refactored code
st.download_button(
label="Download Refactored Code",
data=refactored_code,
file_name="refactored_code.txt",
mime="text/plain",
)
st.success("You can download the refactored code as refactored_code.txt")
# Button to trigger code feedback
if st.button("Get Code Feedback") and code:
with st.spinner("Getting feedback on your code..."):
feedback = code_feedback(code)
st.subheader("Code Feedback:")
st.write(feedback)
# Ensure feedback is a string for download
feedback_text = feedback if isinstance(feedback, str) else str(feedback)
# Provide download option for code feedback
st.download_button(
label="Download Code Feedback",
data=feedback_text, # Use the extracted string here
file_name="code_feedback.txt",
mime="text/plain",
)
st.success("You can download the code feedback as code_feedback.txt")
# Add button to suggest best practices
if st.button("Suggest Best Practices") and code:
with st.spinner("Getting best practices..."):
best_practices = suggest_best_practices(code)
st.subheader("Best Practices Suggestions:")
st.write(best_practices)
# Provide download option for best practices suggestions
best_practices_text = (
best_practices
if isinstance(best_practices, str)
else str(best_practices)
)
st.download_button(
label="Download Best Practices Suggestions",
data=best_practices_text,
file_name="best_practices.txt",
mime="text/plain",
)
st.success(
"You can download the best practices suggestions as best_practices.txt"
)
# Button to trigger error removal
if st.button("Remove Code Errors") and code:
with st.spinner("Removing errors from your code..."):
error_removal_suggestions = remove_code_errors(code)
st.subheader("Error Removal Suggestions:")
st.write(error_removal_suggestions)
# Provide download option for error removal suggestions
error_removal_text = (
error_removal_suggestions
if isinstance(error_removal_suggestions, str)
else str(error_removal_suggestions)
)
st.download_button(
label="Download Error Removal Suggestions",
data=error_removal_text,
file_name="error_removal_suggestions.txt",
mime="text/plain",
)
st.success(
"You can download the error removal suggestions as error_removal_suggestions.txt"
)
if __name__ == "__main__":
main()
|