Enhance text handling in fix_text function and update Mindmap loading message for clarity

image.py

@@ -24,84 +24,81 @@ def fix_base64_padding(data):
     return data
 
 def generate_image(title, category, summary):
-    print("Generating image...")
-    import time
-    start = time.time()
-    try:
-        try:
-            try:
-                negative="low quality, blurry, pixelated, bad anatomy, bad hands, three hands, three legs, bad arms, missing legs, missing arms, poorly drawn face, poorly rendered hands, bad face, fused face, cloned face, worst face, three crus, extra crus, fused crus, worst feet, three feet, fused feet, fused thigh, three thigh, extra thigh, worst thigh, missing fingers, extra fingers, ugly fingers, long fingers, bad composition, horn, extra eyes, huge eyes, 2girl, amputation, disconnected limbs, cartoon, cg, 3d, unreal, animate, cgi, render, artwork, illustration, 3d render, cinema 4d, artstation, octane render, mutated body parts, painting, oil painting, 2d, sketch, bad photography, bad photo, deviant art, aberrations, abstract, anime, black and white, collapsed, conjoined, creative, drawing, extra windows, harsh lighting, jpeg artifacts, low saturation, monochrome, multiple levels, overexposed, oversaturated, photoshop, rotten, surreal, twisted, UI, underexposed, unnatural, unreal engine, unrealistic, video game, deformed body features, NSFW, NUDE, vulgar, negative, unsuitable, inappropriate, offensive, revealing, sexual, explicit",
-                prompt = f"Generate accurate image representing the {category} concept: ```{title.strip()}: {summary.strip()}```"
-                client = Client(
-                    image_provider=RetryProvider(
-                        providers=[ReplicateHome, Airforce],
-                        shuffle=True,
-                        single_provider_retry=True,
-                        max_retries=3,
-                    )
-                )
-                img_data = client.images.generate(
-                    model="sdxl",
-                    prompt=prompt,
-                    negative_prompt=negative,
-                    response_format="b64_json",
-                    width=1024,
-                    height=576,
-                ).data[0].b64_json
-                print(f"Image generated in {time.time() - start:.2f} seconds")
-                if img_data:
-                    return f"data:image/png;base64,{img_data}"
-                return None
-            except Exception as e:
-                print(f"Error generating image: {e}")
-                negative="low quality, blurry, pixelated, bad anatomy, bad hands, three hands, three legs, bad arms, missing legs, missing arms, poorly drawn face, poorly rendered hands, bad face, fused face, cloned face, worst face, three crus, extra crus, fused crus, worst feet, three feet, fused feet, fused thigh, three thigh, extra thigh, worst thigh, missing fingers, extra fingers, ugly fingers, long fingers, bad composition, horn, extra eyes, huge eyes, 2girl, amputation, disconnected limbs, cartoon, cg, 3d, unreal, animate, cgi, render, artwork, illustration, 3d render, cinema 4d, artstation, octane render, mutated body parts, painting, oil painting, 2d, sketch, bad photography, bad photo, deviant art, aberrations, abstract, anime, black and white, collapsed, conjoined, creative, drawing, extra windows, harsh lighting, jpeg artifacts, low saturation, monochrome, multiple levels, overexposed, oversaturated, photoshop, rotten, surreal, twisted, UI, underexposed, unnatural, unreal engine, unrealistic, video game, deformed body features, NSFW, NUDE, vulgar, negative, unsuitable, inappropriate, offensive, revealing, sexual, explicit",
-                prompt = f"Generate accurate image representing the {category} concept: ```{title.strip()}: {summary.strip()}```"
-                client = Client(
-                    image_provider=RetryProvider(
-                        providers=[Airforce, PollinationsAI, BlackboxCreateAgent],
-                        shuffle=True,
-                        single_provider_retry=True,
-                        max_retries=3,
-                    )
-                )
-                img_data = client.images.generate(
-                    model="flux",
-                    prompt=prompt,
-                    negative_prompt=negative,
-                    response_format="b64_json",
-                    width=1024,
-                    height=576,
-                ).data[0].b64_json
-                print(f"Image generated in {time.time() - start:.2f} seconds")
-                if img_data:
-                    return f"data:image/png;base64,{img_data}"
-                return None
-        except Exception as e:
-            print(f"Error generating image: {e}")
-            negative="low quality, blurry, pixelated, bad anatomy, bad hands, three hands, three legs, bad arms, missing legs, missing arms, poorly drawn face, poorly rendered hands, bad face, fused face, cloned face, worst face, three crus, extra crus, fused crus, worst feet, three feet, fused feet, fused thigh, three thigh, extra thigh, worst thigh, missing fingers, extra fingers, ugly fingers, long fingers, bad composition, horn, extra eyes, huge eyes, 2girl, amputation, disconnected limbs, cartoon, cg, 3d, unreal, animate, cgi, render, artwork, illustration, 3d render, cinema 4d, artstation, octane render, mutated body parts, painting, oil painting, 2d, sketch, bad photography, bad photo, deviant art, aberrations, abstract, anime, black and white, collapsed, conjoined, creative, drawing, extra windows, harsh lighting, jpeg artifacts, low saturation, monochrome, multiple levels, overexposed, oversaturated, photoshop, rotten, surreal, twisted, UI, underexposed, unnatural, unreal engine, unrealistic, video game, deformed body features, NSFW, NUDE, vulgar, negative, unsuitable, inappropriate, offensive, revealing, sexual, explicit",
-            prompt = f"Generate accurate image representing the {category} concept: ```{title.strip()}: {summary.strip()}```"
-            client = Client(
-                image_provider=RetryProvider(
-                    providers=[BlackForestLabsFlux1Schnell, VoodoohopFlux1Schnell],
-                    shuffle=True,
-                    single_provider_retry=True,
-                    max_retries=3,
-                )
-            )
-            img_data = client.images.generate(
-                model="flux-schnell",
-                prompt=prompt,
-                negative_prompt=negative,
-                response_format="b64_json",
-                width=1024,
-                height=576,
-            ).data[0].b64_json
-            print(f"Image generated in {time.time() - start:.2f} seconds")
-            if img_data:
-                return f"data:image/png;base64,{img_data}"
-            return None
-    except Exception as e:
-        print(f"Error generating image: {e}")
+        print("Generating image...")
+        import time
+        start = time.time()
+    # try:
+    #     try:
+    #         try:
+    #             negative="low quality, blurry, pixelated, bad anatomy, bad hands, three hands, three legs, bad arms, missing legs, missing arms, poorly drawn face, poorly rendered hands, bad face, fused face, cloned face, worst face, three crus, extra crus, fused crus, worst feet, three feet, fused feet, fused thigh, three thigh, extra thigh, worst thigh, missing fingers, extra fingers, ugly fingers, long fingers, bad composition, horn, extra eyes, huge eyes, 2girl, amputation, disconnected limbs, cartoon, cg, 3d, unreal, animate, cgi, render, artwork, illustration, 3d render, cinema 4d, artstation, octane render, mutated body parts, painting, oil painting, 2d, sketch, bad photography, bad photo, deviant art, aberrations, abstract, anime, black and white, collapsed, conjoined, creative, drawing, extra windows, harsh lighting, jpeg artifacts, low saturation, monochrome, multiple levels, overexposed, oversaturated, photoshop, rotten, surreal, twisted, UI, underexposed, unnatural, unreal engine, unrealistic, video game, deformed body features, NSFW, NUDE, vulgar, negative, unsuitable, inappropriate, offensive, revealing, sexual, explicit",
+    #             prompt = f"Generate accurate image representing the {category} concept: ```{title.strip()}: {summary.strip()}```"
+    #             client = Client()
+    #             img_data = client.images.generate(
+    #                 provider=RetryProvider(
+    #                     providers=[ReplicateHome, Airforce],
+    #                     shuffle=True,
+    #                     single_provider_retry=True,
+    #                     max_retries=3,
+    #                 ),
+    #                 model="sdxl",
+    #                 prompt=prompt,
+    #                 negative_prompt=negative,
+    #                 response_format="b64_json",
+    #                 width=1024,
+    #                 height=576,
+    #             ).data[0].b64_json
+    #             print(f"Image generated in {time.time() - start:.2f} seconds")
+    #             if img_data:
+    #                 return f"data:image/png;base64,{img_data}"
+    #             return None
+    #         except Exception as e:
+    #             print(f"Error generating image: {e}")
+    #             negative="low quality, blurry, pixelated, bad anatomy, bad hands, three hands, three legs, bad arms, missing legs, missing arms, poorly drawn face, poorly rendered hands, bad face, fused face, cloned face, worst face, three crus, extra crus, fused crus, worst feet, three feet, fused feet, fused thigh, three thigh, extra thigh, worst thigh, missing fingers, extra fingers, ugly fingers, long fingers, bad composition, horn, extra eyes, huge eyes, 2girl, amputation, disconnected limbs, cartoon, cg, 3d, unreal, animate, cgi, render, artwork, illustration, 3d render, cinema 4d, artstation, octane render, mutated body parts, painting, oil painting, 2d, sketch, bad photography, bad photo, deviant art, aberrations, abstract, anime, black and white, collapsed, conjoined, creative, drawing, extra windows, harsh lighting, jpeg artifacts, low saturation, monochrome, multiple levels, overexposed, oversaturated, photoshop, rotten, surreal, twisted, UI, underexposed, unnatural, unreal engine, unrealistic, video game, deformed body features, NSFW, NUDE, vulgar, negative, unsuitable, inappropriate, offensive, revealing, sexual, explicit",
+    #             prompt = f"Generate accurate image representing the {category} concept: ```{title.strip()}: {summary.strip()}```"
+    #             client = Client()
+    #             img_data = client.images.generate(
+    #                 provider=RetryProvider(
+    #                     providers=[Airforce, PollinationsAI, BlackboxCreateAgent],
+    #                     shuffle=True,
+    #                     single_provider_retry=True,
+    #                     max_retries=3,
+    #                 ),
+    #                 model="flux",
+    #                 prompt=prompt,
+    #                 negative_prompt=negative,
+    #                 response_format="b64_json",
+    #                 width=1024,
+    #                 height=576,
+    #             ).data[0].b64_json
+    #             print(f"Image generated in {time.time() - start:.2f} seconds")
+    #             if img_data:
+    #                 return f"data:image/png;base64,{img_data}"
+    #             return None
+    #     except Exception as e:
+    #         print(f"Error generating image: {e}")
+    #         negative="low quality, blurry, pixelated, bad anatomy, bad hands, three hands, three legs, bad arms, missing legs, missing arms, poorly drawn face, poorly rendered hands, bad face, fused face, cloned face, worst face, three crus, extra crus, fused crus, worst feet, three feet, fused feet, fused thigh, three thigh, extra thigh, worst thigh, missing fingers, extra fingers, ugly fingers, long fingers, bad composition, horn, extra eyes, huge eyes, 2girl, amputation, disconnected limbs, cartoon, cg, 3d, unreal, animate, cgi, render, artwork, illustration, 3d render, cinema 4d, artstation, octane render, mutated body parts, painting, oil painting, 2d, sketch, bad photography, bad photo, deviant art, aberrations, abstract, anime, black and white, collapsed, conjoined, creative, drawing, extra windows, harsh lighting, jpeg artifacts, low saturation, monochrome, multiple levels, overexposed, oversaturated, photoshop, rotten, surreal, twisted, UI, underexposed, unnatural, unreal engine, unrealistic, video game, deformed body features, NSFW, NUDE, vulgar, negative, unsuitable, inappropriate, offensive, revealing, sexual, explicit",
+    #         prompt = f"Generate accurate image representing the {category} concept: ```{title.strip()}: {summary.strip()}```"
+    #         client = Client()
+    #         img_data = client.images.generate(
+    #             provider=RetryProvider(
+    #                 providers=[BlackForestLabsFlux1Schnell, VoodoohopFlux1Schnell],
+    #                 shuffle=True,
+    #                 single_provider_retry=True,
+    #                 max_retries=3,
+    #             ),
+    #             model="flux-schnell",
+    #             prompt=prompt,
+    #             negative_prompt=negative,
+    #             response_format="b64_json",
+    #             width=1024,
+    #             height=576,
+    #         ).data[0].b64_json
+    #         print(f"Image generated in {time.time() - start:.2f} seconds")
+    #         if img_data:
+    #             return f"data:image/png;base64,{img_data}"
+    #         return None
+    # except Exception as e:
+    #     print(f"Error generating image: {e}")
         return None
 
 def verify_image(image_data):

main.py

@@ -18,7 +18,7 @@ def fix_text(text):
             fixed_text += word.encode('latin1').decode('utf-8')+" "
         except:
             fixed_text += word+" "
-    return fixed_text.encode('utf-8').decode().replace('\\', '\\\\')
+    return fixed_text.encode('utf-8').decode().replace("\n", "\\n").replace("\\n", "\\\\n").replace('\\', '\\\\')
 
 def paper_data(paper_data, wait_time=5):
     data = {"status": "success"}
@@ -42,13 +42,13 @@ def paper_data(paper_data, wait_time=5):
                     continue
                 try:
                     title = fix_text(title)
-                    summary = fix_text(summary)
-                    mindmap = fix_text(mindmap)
                     citation = fix_text(citation)
                     title = html.escape(str(title).strip())
                     citation = html.escape(str(citation).strip())
                     summary = html.escape(str(summary).strip())
+                    print(summary)
                     mindmap = html.escape(str(mindmap).strip())
+                    print(mindmap)
                     status = post_blog(doi, title, category, summary, mindmap, citation, access_key, wait_time)
                 except Exception as e:
                     print(f"Error posting blog '{title}': {e}")

post_blog.py

@@ -31,8 +31,7 @@ def generate_post_html(doi, title, category, summary, mindmap, citation):
         </div>
         <br>
         <h2>Mindmap</h2>
-        <p><small><em>If MindMap doesn't load, go to the <a href="/">Homepage</a> and visit blog again or <a
-                        href="/#">Switch to Android App (Under Development)</a>.</em></small></p>
+        <p><small><em>If MindMap doesn't load, please try refreshing the page.</em></small></p>
         <div class="markmap" id="paper_mindmap">
             <script type="text/template">
                     {mindmap.replace("&amp;", "&").replace(":", "=>").strip()}
@@ -116,10 +115,10 @@ def create_post(doi, title, category, summary, mindmap, citation):
     post_category = f"{category}"
     try:
         post_body, post_image = generate_post_html(doi, title, category, summary, mindmap, citation)
-        # print("_____________________\n\n",title,"\n\n_____________________")
-        # with open(f'{title.replace(" ", "_")}.html', 'w', encoding='utf-8') as f:
-        #     f.write(post_body)
-        # exit()
+        print("_____________________\n\n",title,"\n\n_____________________")
+        with open(f'{title.replace(" ", "_")}.html', 'w', encoding='utf-8') as f:
+            f.write(post_body)
+        exit()
     except Exception as e:
         print(f"Error generating post: {e}")
         return None, None, None, None