Update app.py

app.py

@@ -59,7 +59,10 @@ class CodeGenerationTool(Tool):
         # This is a simplified example, real implementation would use a code generation model
         language = arguments.get("language", "python")
         code = arguments.get("code", "print('Hello, World!')")
-        return {"output": f"```{language}\n{code}\n```"}
+        return {"output": f"
+
+
+{language}\n{code}\n
 
 class DataRetrievalTool(Tool):
     def __init__(self):
@@ -366,7 +369,7 @@ if st.session_state.chat_state == "normal":
                 model = RetroRecodeModel()
             else:
                 st.error("Invalid model selection.")
-                return
+                st.stop()  # Stop the Streamlit app execution
 
             response = model.generate_text(formated_text, temperature=temp_values)
             st.markdown(response)