Spaces:
Sleeping
Sleeping
| 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() | |