Update app.py

app.py

@@ -6,6 +6,7 @@ from nltk.corpus import wordnet
 import nltk
 nltk.download('wordnet')
 
+
 def generate_text(prompt, temperature=0.7, top_k=50, repetition_penalty=1.2, max_length=None, min_length=10):
     text_model = "gpt2"
     tokenizer = AutoTokenizer.from_pretrained(text_model)
@@ -80,37 +81,26 @@ try :
                 new_image1 = generated_image1.resize((700, 500))
                 st.image(new_image1, use_column_width=True)
 
+
+            # Introduction
+            st.subheader("Introduction")
             intro_text = generate_text(f'introduction about : {generated_text}', min_length=100, max_length=200)
             intro_text = intro_text.replace(f"introduction about : {generated_text}", "")
             st.write(intro_text.strip())  # Display the generated introduction text
 
-            body_text1 = generate_text(f'article about : {generated_text}', min_length=100, max_length=150)
-            body_text1 = body_text1.replace(f"article about : {generated_text}", "")
-            st.write(body_text1.strip())  # Display the generated introduction text
-
-            body_text2 = generate_text(f'article about : {generated_text}', min_length=200, max_length=300)
-            body_text2 = body_text2.replace(f"{generated_text}", "")
-            st.write(body_text2.strip())  # Display the generated introduction text
-
-            conclusion_text = generate_text(f'conclusion about : {generated_text}', min_length=100, max_length=200)
-            conclusion_text = conclusion_text.replace(f"conclusion about : {generated_text}", "")
-            st.write(conclusion_text.strip())  # Display the generated introduction text
-
-
-
-            st.subheader("Introduction")
-            st.write(intro_text)
             modified_prompt = generated_text + 'bright'
             generated_image2 = generate_image(modified_prompt)
 
             new_image2 = generated_image2.resize((700, 300))
             st.image(new_image2, use_column_width=True)
 
-
+            # Body 1
             col1, col2 = st.columns(2)
             with col1:
                 st.subheader("Body")
-                st.write(body_text1)
+                body_text1 = generate_text(f'article about : {generated_text}', min_length=100, max_length=150)
+                body_text1 = body_text1.replace(f"article about : {generated_text}", "")
+                st.write(body_text1.strip())  # Display the generated introduction text
 
             with col2:
                 modified_prompt2 = generated_text + 'shade'
@@ -118,14 +108,22 @@ try :
                 st.markdown("<br><br><br><br>", unsafe_allow_html=True)
                 st.image(generated_image3, use_column_width=True)
 
-            st.write(body_text2)
+            # Body 2
+            body_text2 = generate_text(f'article about : {generated_text}', min_length=200, max_length=300)
+            body_text2 = body_text2.replace(f"{generated_text}", "")
+            st.write(body_text2.strip())  # Display the generated introduction text
+
             modified_prompt3 = generated_text + title
             generated_image4 = generate_image(modified_prompt3)
             new_image3 = generated_image4.resize((700, 300))
             st.image(new_image3, use_column_width=True)
 
+            # Conclusion
             st.subheader("Conclusion")
-            st.write(conclusion_text)
+            conclusion_text = generate_text(f'conclusion about : {generated_text}', min_length=100, max_length=200)
+            conclusion_text = conclusion_text.replace(f"conclusion about : {generated_text}", "")
+            st.write(conclusion_text.strip())  # Display the generated introduction text
+
         else:
             st.warning("Please input keywords to generate content.")