Update app.py

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import ast
+import astor
 import os
 
 # Initialize Hugging Face model and tokenizer
@@ -25,32 +27,41 @@ def analyze_code(code):
     
     return results
 
-# Function to simulate quality and bug detection
-# Note: Replace with a real analysis pipeline.
-def check_code_quality_and_bugs(code):
+# Function to detect and fix bugs
+def detect_and_fix_bugs(code):
     suggestions = []
+    fixed_code = code
 
-    # Example: Check for readability (placeholder logic)
-    if len(code.split("\n")) < 5:
-        suggestions.append("Code seems too short, ensure functionality is implemented correctly.")
+    try:
+        tree = ast.parse(code)
 
-    # Example: Check for comments (placeholder logic)
-    if "#" not in code:
-        suggestions.append("Consider adding comments to improve code clarity.")
+        # Example: Check for missing except blocks in try statements
+        for node in ast.walk(tree):
+            if isinstance(node, ast.Try):
+                if not any(isinstance(handler, ast.ExceptHandler) for handler in node.handlers):
+                    suggestions.append("Add an 'except' block to handle exceptions in the 'try' statement.")
+                    # Add a generic except block as a fix
+                    new_handler = ast.ExceptHandler(type=None, name=None, body=[ast.Pass()])
+                    node.handlers.append(new_handler)
 
-    # Example: Check for style guide adherence (placeholder logic)
-    if "import" in code and "os" not in code:
-        suggestions.append("Unused imports detected; consider removing them.")
+        # Example: Check for missing comments
+        if not any(isinstance(node, ast.Expr) and isinstance(node.value, ast.Str) for node in tree.body):
+            suggestions.append("Consider adding a module-level docstring or comments to improve code clarity.")
+            # Add a module-level docstring as a fix
+            docstring = ast.Expr(value=ast.Str(s="""Add a description of the module here."""))
+            tree.body.insert(0, docstring)
 
-    # Example: Detect code smells (placeholder logic)
-    if "try" in code and "except" not in code:
-        suggestions.append("`try` block without `except` may lead to unhandled exceptions.")
+        # Convert the modified AST back to code
+        fixed_code = astor.to_source(tree)
 
-    return suggestions
+    except Exception as e:
+        suggestions.append(f"Error analyzing code: {e}")
+
+    return suggestions, fixed_code
 
 # Streamlit app UI
-st.title("Code Quality and Bug Detection Tool")
-st.markdown("Analyze your code for syntax issues, quality, and potential bugs.")
+st.title("Code Quality, Bug Detection, and Auto-Correction Tool")
+st.markdown("Analyze your code for syntax issues, quality, bugs, and get suggested corrections.")
 
 # File uploader
 uploaded_file = st.file_uploader("Upload a Python code file", type=["py"])
@@ -58,7 +69,7 @@ uploaded_file = st.file_uploader("Upload a Python code file", type=["py"])
 # Code snippet input
 code_snippet = st.text_area("Or paste your code snippet below:")
 
-if st.button("Analyze Code"):
+if st.button("Analyze and Fix Code"):
     if uploaded_file is not None:
         code = uploaded_file.read().decode("utf-8")
     elif code_snippet.strip():
@@ -67,16 +78,20 @@ if st.button("Analyze Code"):
         st.error("Please upload a file or paste code to analyze.")
         st.stop()
 
-    # Perform code analysis
+    # Perform code analysis and bug fixing
     st.subheader("Analysis Results")
     
     st.write("**Code Quality and Bug Suggestions:**")
-    suggestions = check_code_quality_and_bugs(code)
+    suggestions, fixed_code = detect_and_fix_bugs(code)
     if suggestions:
         for i, suggestion in enumerate(suggestions, 1):
             st.write(f"{i}. {suggestion}")
     else:
-        st.write("No issues detected. Your code looks good!")
+        st.write("No major issues detected. Your code looks good!")
+
+    # Display corrected code
+    st.subheader("Corrected Code:")
+    st.code(fixed_code, language="python")
 
     # Simulated CodeBERT analysis (placeholder)
     st.write("**Model Analysis:**")
@@ -86,3 +101,4 @@ if st.button("Analyze Code"):
 
 st.markdown("---")
 st.markdown("*Powered by Hugging Face and Streamlit*")
+