Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import streamlit as st
 from langchain.prompts import PromptTemplate
 import requests
@@ -6,13 +7,9 @@ from langchain.chains import LLMChain
 import io
 from PIL import Image
 import json
-from model import create_model
-import torch
-api_key = st.secrets["API"]
-
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+from model import model,tokenizer
 
-model,tokenizer = create_model()
+API = st.secrets("API")
 # Load existing ideas from a file
 def load_ideas():
     try:
@@ -83,7 +80,7 @@ def divide_content(text):
 
 # Image Generation
 API_URL = "https://api-inference.huggingface.co/models/goofyai/3d_render_style_xl"
-headers = {"Authorization": f"Bearer {api_key}"}
+headers = {"Authorization": "Bearer API"}
 
 def query(payload):
 	response = requests.post(API_URL, headers=headers, json=payload)
@@ -97,37 +94,33 @@ def generat_image(image_prompt,name):
       image.save(f"{name}.png")
       return image
 
-# st.title('AI Blog Content Generator')
-
-# st.header('What is the topic of your blog post?')
-# topic_text = st.text_input('Topic:', placeholder='Example: Generative AI')
-# submitted = st.button('Submit')
-
 def display_content_with_images(blog):
+    blog_images = [key for key in list(blog.keys()) if "_image" in key]
     # Streamlit Display
     st.header(blog['title'])
-
+    i = 0
     # Introduction
     col1, col2 = st.columns(2, gap='medium')
     with col1:
       st.header('Introduction')
       st.write(blog['Introduction'])
     with col2:
-      st.image(blog['introduction_image'], use_column_width=True)
+      st.image(blog[blog_images[i]], use_column_width=True)
+      i+=1
 
     # History
     st.header('History and Background')
     st.write(blog['History and Background'])
-    st.image(blog['History and Background_image'], use_column_width=True)
-
+    st.image(blog[blog_images[i]], use_column_width=True)
+    i+=1
     # Content
     col1, col2 = st.columns(2, gap='medium')
     with col1:
       st.header('Key Concepts and Terminology')
       st.write(blog['Key Concepts and Terminology'])
     with col2:
-      st.image(blog['Key Concepts and Terminology_image'], use_column_width=True)
-
+      st.image(blog[blog_images[i]], use_column_width=True)
+      i+=1
     # Use Cases and Applications
     st.header('Use Cases and Applications')
     st.write(blog['Use Cases and Applications'])
@@ -146,8 +139,8 @@ def display_content_with_images(blog):
       st.header('Conclusion')
       st.write(blog['Conclusion'])
     with col2:
-      st.image(blog['Conclusion_image'])
-
+      st.image(blog[blog_images[i]], use_column_width=True)
+      i+=1
 
 
 # Streamlit App
@@ -190,75 +183,33 @@ if button_clicked and topic not in keys:
     blog = divide_content(content)
     st.write(blog)
     st.header(topic)
-    # Introduction
-    col1, col2 = st.columns(2, gap='medium')
-    with col1:
-      st.subheader('Introduction')
-      st.write(blog[' Introduction'])
-    with col2:
-      image_prompt = blog[' Introduction'].splitlines()[0]
-      introduction_image = generat_image(image_prompt,f" Introduction{topic}")
-      st.image(introduction_image, use_column_width=True)
-
-    st.divider()
-
-    # History
-    st.subheader('History and Background')
-    st.write(blog[' History and Background'])
-    image_prompt = blog[' History and Background'].splitlines()[0]
-    history_image =generat_image(image_prompt,f" History and Background{topic}")
-    st.image(history_image, use_column_width=True)
-    st.divider()
-
-    # Key Concepts and Terminology
-    col1, col2 = st.columns(2, gap='medium')
-    with col1:
-      st.subheader('Key Concepts and Terminology')
-      st.write(blog[' Key Concepts and Terminology'])
-    with col2:
-      image_prompt = blog[' Key Concepts and Terminology'].splitlines()[0]
-      key_concepts_image = generat_image(image_prompt,f" Key Concepts and Terminology{topic}")
-      st.image(key_concepts_image, use_column_width=True)
-    st.divider()
-
-    # Use Cases and Applications
-    st.subheader('Use Cases and Applications')
-    st.write(blog[' Use Cases and Applications'])
-
-    # Benefits and Drawbacks
-    st.subheader('Benefits and Drawbacks')
-    st.write(blog[' Benefits and Drawbacks'])
-
-    # Future Outlook
-    st.subheader('Future Outlook')
-    st.write(blog[' Future Outlook'])
-
-    # Conclusion
-    col1, col2 = st.columns(2, gap='medium')
-    with col1:
-      st.subheader('Conclusion')
-      st.write(blog[' Conclusion'])
-    with col2:
-      image_prompt = blog[' Conclusion'].splitlines()[0]
-      conclusion_image = generat_image(image_prompt,f" Conclusion{topic}")
-      st.image(conclusion_image, use_column_width=True)
-
+    keyss = list(blog.keys())
+    image_prompts = []
+    i=0
+    while len(image_prompts)<4:
+      try:
+        image_prompts.append((keyss[i],blog[keyss[i]].splitlines()[0]))
+        i+=1
+      except Exception as e:
+        print(e)
+        i+=1
 
     # Blog Data
     blog_data = {
         'title': topic,
         'Introduction': blog[' Introduction'],
-        'introduction_image': f" Introduction{topic}.png",
         'History and Background': blog[' History and Background'],
-        'History and Background_image':  f" History and Background{topic}.png",
         'Key Concepts and Terminology': blog[' Key Concepts and Terminology'],
-        'Key Concepts and Terminology_image': f" Key Concepts and Terminology{topic}.png",
         'Use Cases and Applications': blog[' Use Cases and Applications'],
         'Benefits and Drawbacks': blog[' Benefits and Drawbacks'],
         'Future Outlook': blog[' Future Outlook'],
         'Conclusion': blog[' Conclusion'],
-        'Conclusion_image': f" Conclusion{topic}.png"
     }
+    for k,image in image_prompts:
+      img = generat_image(image,f" {k}{topic}")
+      blog_data[f'{k}_image'] = f" {k}{topic}.png"
+    
+    display_content_with_images(blog_data)
 
     # Save blog with images
     existing_ideas.append({topic: blog_data})