Create app.py

app.py

@@ -0,0 +1,97 @@
+import gradio as gr
+import requests
+import json
+from PIL import Image
+import io
+import base64
+
+API_URL = "https://genv01.plutovid.com/get_image"
+COUNT_URL = "https://genv01.plutovid.com/get_image_count"
+
+def get_total_generated_count():
+    try:
+        response = requests.get(COUNT_URL)
+        response.raise_for_status()
+        return f"Total Images Generated So Far: {response.json().get('count', 0)}"
+    except requests.exceptions.RequestException:
+        return "Could not fetch generation count."
+
+def generate_from_hosted_api(prompt, progress=gr.Progress(track_tqdm=True)):
+    if not prompt:
+        raise gr.Error("Prompt cannot be empty.")
+
+    images = []
+    
+    try:
+        response = requests.get(API_URL, params={'message': prompt}, stream=True)
+        response.raise_for_status()
+
+        for line in response.iter_lines():
+            if line:
+                decoded_line = line.decode('utf-8')
+                if decoded_line.startswith('event: status'):
+                    pass
+                elif decoded_line.startswith('data:'):
+                    try:
+                        data_json = json.loads(decoded_line[len('data:'):])
+                        if 'message' in data_json:
+                            progress(0.5, desc=data_json['message'])
+                        elif 'images' in data_json:
+                            progress(0.9, desc="Decoding images...")
+                            for b64_image in data_json['images']:
+                                img_data = base64.b64decode(b64_image)
+                                img = Image.open(io.BytesIO(img_data))
+                                images.append(img)
+                    except json.JSONDecodeError:
+                        pass
+
+    except requests.exceptions.RequestException as e:
+        raise gr.Error(f"Failed to connect to the generation service: {e}")
+    
+    if not images:
+        return None, "Generation failed or no images were returned."
+
+    return images, "Generation complete!"
+
+
+title = "PlutoGen V0.1"
+description = """
+# Welcome to PlutoGen V0.1
+
+**Crafting Realistic and Non-Detectable AI Content**
+
+This is the first model to be released open-source in the near future by **[PlutoLabs](https://huggingface.co/plutolabs)**, a non-profit from Russia dedicated to keeping pace with current technology. Our goal with PlutoGen is to push the boundaries of AI content generation, making it virtually indistinguishable from human-created content.
+
+While this very first version may not seem like much, it's a step towards infinite possibilities. If you like this project, please consider leaving a like or following our organization for future releases and the upcoming open-source release of v0.1!
+
+Enter a prompt below to see what PlutoGen can do.
+"""
+
+with gr.Blocks(theme='soft') as demo:
+    gr.Markdown(f"<h1 style='text-align: center;'>{title}</h1>")
+    gr.Markdown(description)
+    
+    with gr.Row():
+        total_count_output = gr.Textbox(label="Generation Stats", interactive=False)
+
+    with gr.Row():
+        with gr.Column(scale=2):
+            prompt_input = gr.Textbox(label="Prompt", placeholder="e.g., a photorealistic portrait of a cat wearing a tiny hat")
+            submit_button = gr.Button("Generate Images", variant="primary")
+            status_output = gr.Textbox(label="Status", interactive=False)
+        with gr.Column(scale=3):
+            image_output = gr.Gallery(label="Generated Images", columns=2, height="auto")
+
+    submit_button.click(
+        fn=generate_from_hosted_api,
+        inputs=[prompt_input],
+        outputs=[image_output, status_output]
+    ).then(
+        fn=get_total_generated_count,
+        outputs=total_count_output
+    )
+
+    demo.load(get_total_generated_count, None, total_count_output)
+
+if __name__ == "__main__":
+    demo.launch()