Upload app.py

app.py

@@ -62,17 +62,13 @@ def generate_webapp():
             prompt_UML = "Escreva somente o script necessário (não coloque nenhum comentário nada a mais do que somente o script) usando plantuml para geração do digrama de classes tendo como base o seguinte código:"+generated_code.text
             # Gerar código usando o modelo gemini-pro
             model_code = genai.GenerativeModel('gemini-pro')
-            generated_code = model_code.generate_content(prompt_UML)
+            generated_code = model_code.generate_content(prompt_UML)         
 
-            # Exibir o código gerado no frontend
-            st.subheader("Código Gerado:")
-            st.code(generated_code.text, language='python')
-
-            generated_code = generated_code.text
-            print(generated_code)
+            uml_code = generated_code.text
+            print(uml_code)
             with st.spinner("Gerando o diagrama de classes da app..."):
                 # Gera o diagrama e salva como um arquivo PNG
-                image_path = plantuml.processes(generated_code)
+                image_path = plantuml.processes(uml_code)
                 with open("class_diagram.png", "wb") as f:
                     f.write(image_path)