Remove summary and mindmap fields from test function for cleaner data structure

image.py

@@ -6,7 +6,7 @@ import requests
 from urllib.parse import quote
 from PIL import Image
 from g4f.client import Client
-from g4f.Provider import RetryProvider, PollinationsAI, Airforce
+from g4f.Provider import RetryProvider, PollinationsAI, Airforce, ReplicateHome, BlackboxCreateAgent
 from g4f.Provider.hf_space.BlackForestLabsFlux1Schnell import BlackForestLabsFlux1Schnell
 from g4f.Provider.hf_space.VoodoohopFlux1Schnell import VoodoohopFlux1Schnell
 
@@ -23,35 +23,88 @@ def fix_base64_padding(data):
         data += "=" * (4 - missing_padding)
     return data
 
-def generate_image(title, category):
+def generate_image(title, category, summary):
     print("Generating image...")
     import time
     start = time.time()
     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 = quote(f"Generate accurate image representing the concept: ```((( {category}))) {title.strip() }```")
-        client = Client(
-            image_provider=RetryProvider(
-                providers=[BlackForestLabsFlux1Schnell, VoodoohopFlux1Schnell],
-                shuffle=True,
-                single_provider_retry=True,
-                max_retries=3,
+        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()}```"
+                print(prompt)
+                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()}```"
+                print(prompt)
+                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()}```"
+            print(prompt)
+            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
+            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"An error occurred during image generation: {e}")
+        print(f"Error generating image: {e}")
         return None
 
 def verify_image(image_data):
@@ -81,12 +134,13 @@ def upload_image(data_uri, api_key):
     finally:
         return image_url
 
-def fetch_image(title, category, api_key):
+def fetch_image(title, category, summary, api_key):
     title = r"{}".format(title)
     category = r"{}".format(category)
+    summary = extract_summary(summary)
     image_url = "https://i.ibb.co/TBJqggw/Image-Not-Found.jpg"
     try:
-        data_uri = generate_image(title, category)
+        data_uri = generate_image(title, category, summary)
         if data_uri:
             base64_image = fix_base64_padding(data_uri.split(",")[1])
             image_data = None
@@ -109,8 +163,9 @@ def fetch_image(title, category, api_key):
 
 
 if __name__ == "__main__":
-    title = "Accelerated cell-type-specific regulatory evolution of the Homo sapiens brain"
-    category = "This study investigates the accelerated regulatory evolution of gene expression in human brain cell types compared to chimpanzees. It reveals significant differences in gene expression, particularly in excitatory and inhibitory neurons, highlighting the role of regulatory evolution in human cognitive and behavioral traits"
+    title = "Exposition: Enumerative Geometry and Tree-Level Gromov-Witten Invariants"
+    category = "Mathematics"
+    summary = "The text discusses the Kontsevich-Manin formula for enumerating degree d rational curves via Gromov-Witten invariants. It details the calculation of these invariants using moduli spaces of stable maps and explores their implications in enumerative geometry."
     api_key = "aa38b04047587c609f5c7e22f9d840f0"
-    image_url = fetch_image(title, category, api_key)
+    image_url = fetch_image(title, category, summary, api_key)
     print(image_url)

main.py

@@ -83,8 +83,6 @@ def test(uaccess_key):
                 "category":"Economics",
                 "pdf_url":"https://arxiv.org/pdf/2501.00578",
                 "citation":"Miller, A. D. (2025). The limits of tolerance (Version 1). arXiv. https://doi.org/10.48550/ARXIV.2501.00578",
-                "summary":"## Summary\nThe text discusses the concept of community standards in relation to good faith terminations and the criminalization of obscene speech in the U.S. It introduces a model for aggregating community judgments represented as intervals, focusing on endpoint rules that can ensure strategyproofness and responsiveness in preference aggregation.\n\n## Highlights\n- Community standards are referenced for good faith terminations.\n- Obscene speech can be criminalized based on contemporary community standards.\n- A model is introduced where judgments are represented as intervals on the real line.\n- The p,q-th endpoint rule aggregates these judgments.\n- Endpoint rules allow for independent aggregation of upper and lower endpoints.\n- Generalized single-peaked preferences are defined for effective aggregation.\n- The maximal rule represents the most liberal approach to community standards.\n\n## Key Insights\n- Community Standards and Good Faith: The reliance on community standards for terminations reflects societal values, which can fluctuate over time, necessitating a flexible approach to legal interpretations.\n- Obscenity and Free Speech: The ability to criminalize obscene speech underlines the tension between individual rights and community morals, highlighting the complexities in defining obscenity.\n- Interval Representation: The model of representing judgments as intervals allows for a more nuanced understanding of community preferences, accommodating diverse opinions while still aiming for consensus.\n- Endpoint Rules: These rules facilitate a structured method for aggregating individual judgments, ensuring that preferences are considered without diminishing the integrity of the decision-making process.\n- Generalized Single-Peaked Preferences: This concept aids in simplifying the aggregation of preferences by establishing a clear peak, which enhances the efficiency of collective decision-making.\n- Strategyproofness in Voting: By ensuring that preferences are aggregated independently, the model maintains fairness and prevents manipulation, reinforcing the integrity of democratic processes.\n- Liberalism vs. Democracy: The discussion draws a connection between endpoint rules and broader political philosophies, suggesting that the balance between individual freedoms and collective decision-making is crucial for effective governance.",
-                "mindmap":"## Terminations of Good Faith\n- Reference to community standards of decency\n- Fairness and reasonableness in terminations\n\n## Obscenity and Free Speech\n- Criminalization based on contemporary community standards\n- Free speech protections may not apply\n\n## Community Standards Model\n- Standards represented by real line intervals\n- Judgment takes the form of intervals\n\n## Endpoint Rules\n- Defined by p-th lowest and q-th highest\n- Aggregate judgments based on endpoints\n\n## Strategyproofness Implications\n- Restrict allowable preferences for strategyproofness\n- Single-peaked preferences enable non-dictatorial rules\n\n## Generalized Single-Peaked Preferences\n- Unique preferred interval called the \"peak\"\n- Intervals between peak and third are preferred\n\n## Aggregation of Endpoints\n- Independent aggregation of upper and lower endpoints\n- Characterized by responsiveness and neutrality axioms\n\n## Median and Maximal Rules\n- Median rule is least permissive, satisfies conditions\n- Maximal rule allows any reasonable action\n\n## Phantom Intervals\n- Include half-bounded and fully unbounded intervals\n- Used in interval aggregation models\n\n## Axioms in Aggregation Methods\n- Seven axioms define endpoint rule characteristics\n- Ensures fair aggregation of individual judgments"
             },
         },
     }