Update app.py

app.py

@@ -49,7 +49,9 @@ if st.button("Generate Diagram"):
                 with open(temp_file, "w") as f:
                     f.write(python_code)
 
-                # Execute the code and look for 'diagram.jpg'
+                # Change directory to temp and execute the code
+                original_dir = os.getcwd()
+                os.chdir(tmpdirname)
                 try:
                     exec(open(temp_file).read())
                     diagram_path = os.path.join(tmpdirname, "diagram.jpg")
@@ -61,6 +63,8 @@ if st.button("Generate Diagram"):
                         st.error("No diagram image was generated.")
                 except Exception as e:
                     st.error(f"Error executing the Python code: {e}")
+                finally:
+                    os.chdir(original_dir)  # Restore the original directory
         else:
             st.write("Response:", output)
     else: