Added the UML image to the README.md

graphene_llm_readme_gen/ReadmeGenModule/templates/index.html

@@ -253,6 +253,7 @@
             <h2>Generated README content</h2>
             <form action="/readme" method="POST">
                 <pre id="generated-readme" style="white-space: pre-wrap;">{{ readme_content }}</pre>
+                <img src="{{ url_for('static', filename='uml_name.png') }}" alt="UML Diagram" width="1500" height="200">
                 <input type="submit" value="Show README">
             </form>
         </div>

graphene_llm_readme_gen/ReadmeGenModule/tools/utils_uml_tool.py

@@ -86,6 +86,8 @@ def run_tool(tool_name, directory_path, llm, query):
         }
         readme_path = os.path.join(directory_path, 'README.md')
         write_to_readme(readme_path, readme_content)
+        uml_name = 'uml_name.png'
+        shutil.copy(img_path + '.png', os.path.join('static', uml_name))
     else:
         print('UML image already exists. Skipping rendering.')