Update app.py

app.py

@@ -49,6 +49,15 @@ def generate_summary(text):
     )
     return response['choices'][0]['message']['content']
 
+# Function to fix bugs in code
+def fix_code_bugs(buggy_code, model="gpt-4o-mini"):
+    prompt = f"The following code has bugs or issues. Please identify and fix the problems. If possible, provide explanations for the changes made.\n\nBuggy Code:\n{buggy_code}\n\nFixed Code:"
+    response = openai.ChatCompletion.create(
+        model=model,
+        messages=[{"role": "user", "content": prompt}]
+    )
+    return response['choices'][0]['message']['content']
+
 # Streamlit app starts here
 st.title("AI Assistance")
 
@@ -58,9 +67,15 @@ openai_api_key = st.text_input("Enter your OpenAI API key:", type="password")
 if openai_api_key:
     openai.api_key = openai_api_key
 
-    # Sidebar to toggle between Course Query Assistant and Code Generator
+    # Sidebar to toggle between Course Query Assistant, Code Generator, Bug Fixer, etc.
     st.sidebar.title("Select Mode")
-    mode = st.sidebar.radio("Choose an option", ("Course Query Assistant", "Code Generator", "AI Chatbot Tutor", "AI Study Notes & Summaries"))
+    mode = st.sidebar.radio("Choose an option", (
+        "Course Query Assistant", 
+        "Code Generator", 
+        "AI Chatbot Tutor", 
+        "AI Study Notes & Summaries", 
+        "Code Bug Fixer"
+    ))
 
     if mode == "Course Query Assistant":
         st.header("Course Query Assistant")
@@ -200,4 +215,33 @@ if openai_api_key:
 
             # Display the summary
             st.write("### AI-Generated Summary:")
-            st.write(summary)
+            st.write(summary)
+
+    elif mode == "Code Bug Fixer":
+        st.header("Code Bug Fixer")
+
+        # User input for buggy code
+        buggy_code = st.text_area("Enter your buggy code here:")
+
+        if st.button("Fix Code"):
+            if buggy_code:
+                with st.spinner("Fixing code..."):
+                    # Fix bugs using GPT-4
+                    fixed_code = fix_code_bugs(buggy_code)
+                    
+                    # Display the fixed code
+                    st.write("### Fixed Code:")
+                    st.code(fixed_code, language="python")
+
+                    # Provide a download link for the fixed code
+                    with open("fixed_code.txt", "w") as f:
+                        f.write(fixed_code)
+
+                    st.download_button(
+                        label="Download Fixed Code",
+                        data=open("fixed_code.txt", "rb").read(),
+                        file_name="fixed_code.txt",
+                        mime="text/plain"
+                    )
+            else:
+                st.error("Please enter some buggy code to fix.")