Update app.py

app.py

@@ -1,12 +1,13 @@
 import gradio as gr
 from agents.error_fixer import fix_code
+from agents.code_modifier import modify_code
 
 def process_code(code, prompt):
     # Placeholder logic – we'll replace this later
-    if prompt.lower().strip() == "":
+    if prompt.strip() == "":
         return fix_code(code)
     else:
-        return fix_code(code) + f"\n\n# Prompt modifications not implemented yet" 
+        return modify_code(code, prompt)  
     
 
 with gr.Blocks() as demo: