Create app.py

app.py

@@ -0,0 +1,192 @@
+import gradio as gr
+import graphviz
+import os
+from PIL import Image, ImageDraw, ImageFont
+
+# --- Setup: Create directories and placeholder image ---
+
+# Create a directory for output images if it doesn't exist
+if not os.path.exists("outputs"):
+    os.makedirs("outputs")
+
+def create_placeholder_image(path="placeholder.png"):
+    """Creates a placeholder image to display on app startup."""
+    try:
+        img = Image.new('RGB', (600, 800), color=(255, 255, 255))
+        draw = ImageDraw.Draw(img)
+        try:
+            # Use a common font, fallback to default
+            font = ImageFont.truetype("DejaVuSans.ttf", 24)
+        except IOError:
+            font = ImageFont.load_default()
+        text = "Flowchart will be generated here"
+        # Get text size using textbbox
+        bbox = draw.textbbox((0, 0), text, font=font)
+        text_width = bbox[2] - bbox[0]
+        text_height = bbox[3] - bbox[1]
+        position = ((600 - text_width) / 2, (800 - text_height) / 2)
+        draw.text(position, text, fill=(200, 200, 200), font=font)
+        img.save(path)
+        return path
+    except Exception:
+        return None
+
+# --- Core Logic: Flowchart Generation ---
+
+def generate_flowchart(prompt: str) -> str:
+    """
+    Generates a flowchart using Graphviz based on the input prompt.
+    Returns the file path of the generated PNG image.
+    """
+    # A simple keyword-based logic to generate different flowcharts
+    if "coffee" in prompt.lower():
+        dot_string = """
+        digraph G {
+            graph [pad="0.5", nodesep="0.5", ranksep="0.8", fontname="helvetica"];
+            node [shape=box, style="rounded,filled", fillcolor="#EAEAFB", fontname="helvetica"];
+            edge [color="#666666"];
+            rankdir=TB;
+
+            start [label="Start", shape=ellipse, fillcolor="#D5D6F9"];
+            have_beans [label="Have coffee beans?", shape=diamond, fillcolor="#F9EED5"];
+            buy_beans [label="Buy coffee beans"];
+            grind_beans [label="Grind beans"];
+            boil_water [label="Boil water"];
+            add_grounds [label="Add grounds to filter"];
+            pour_water [label="Pour water over grounds"];
+            wait [label="Wait for brewing"];
+            pour_cup [label="Pour coffee into cup"];
+            add_extras [label="Add milk/sugar?", shape=diamond, fillcolor="#F9EED5"];
+            add_milk_sugar [label="Add milk and/or sugar"];
+            drink [label="Drink coffee"];
+            end_node [label="Enjoy your coffee!", shape=ellipse, fillcolor="#D5D6F9"];
+
+            start -> have_beans;
+            have_beans -> grind_beans [label="  Yes"];
+            have_beans -> buy_beans [label="No  "];
+            buy_beans -> grind_beans;
+            grind_beans -> boil_water;
+            boil_water -> add_grounds;
+            add_grounds -> pour_water;
+            pour_water -> wait;
+            wait -> pour_cup;
+            pour_cup -> add_extras;
+            add_extras -> add_milk_sugar [label="  Yes"];
+            add_extras -> drink [label="No  "];
+            add_milk_sugar -> drink;
+            drink -> end_node;
+        }
+        """
+    else:
+        # A default flowchart for any other prompt
+        cleaned_prompt = prompt.replace("'", "").replace("\"", "")[:60]
+        dot_string = f"""
+        digraph G {{
+            graph [fontname="helvetica"];
+            node [shape=box, style="rounded,filled", fillcolor="#EAEAFB", fontname="helvetica"];
+            rankdir=TB;
+            
+            start [label="Start", shape=ellipse, fillcolor="#D5D6F9"];
+            process [label="Process user request:\\n'{cleaned_prompt}... '"];
+            end_node [label="End", shape=ellipse, fillcolor="#D5D6F9"];
+            
+            start -> process -> end_node;
+        }}
+        """
+
+    try:
+        graph = graphviz.Source(dot_string)
+        # Save the rendered graph to a file, overwriting previous ones
+        output_path = graph.render(os.path.join("outputs", "flowchart"), format='png', cleanup=True)
+        return output_path
+    except Exception as e:
+        print(f"Error rendering graphviz: {e}")
+        return create_placeholder_image("error.png") # Return an error image if something goes wrong
+
+
+def handle_generate_click(prompt: str):
+    """Function to be called on button click. Updates the UI."""
+    if not prompt:
+        # Handle empty prompt case
+        placeholder_path = create_placeholder_image()
+        return placeholder_path, gr.update(visible=False)
+        
+    image_path = generate_flowchart(prompt)
+    
+    # Return the image path to update the gr.Image component
+    # and a gr.update object for the gr.DownloadButton
+    return image_path, gr.update(value=image_path, visible=True)
+
+
+# --- Gradio UI ---
+
+# Custom CSS to hide the footer for a cleaner look like the screenshot
+css = """
+footer {display: none !important}
+.gradio-container {background-color: #f8f9fa}
+"""
+
+with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
+    gr.Markdown("# AI Flowchart Generator")
+    gr.Markdown(
+        "Our AI Flowchart Generator allows you to create detailed flowcharts instantly. Whether you need a free "
+        "online AI flowchart generator from text or an intuitive flowchart maker AI, this tool delivers accurate and "
+        "visually engaging results. Discover how AI can enhance your workflow with the best flow chart generator "
+        "AI solution available online."
+    )
+
+    with gr.Group(): # Using gr.Group to create the white card effect
+        with gr.Row(equal_height=False):
+            with gr.Column(scale=1):
+                lang_dropdown = gr.Dropdown(
+                    label="Language",
+                    choices=["English"],
+                    value="English"
+                )
+                chart_type_dropdown = gr.Dropdown(
+                    choices=["Flowchart"],
+                    value="Flowchart",
+                    label=" " # Using space for an empty label like in the screenshot
+                )
+                prompt_input = gr.Textbox(
+                    lines=10,
+                    placeholder="Generate a flowchart for making coffee",
+                    value="Generate a flowchart for making coffee",
+                    label=" " # Empty label
+                )
+                with gr.Row():
+                    # These buttons are for UI purposes and are not connected to logic
+                    templates_btn = gr.Button("📄 Templates")
+                    generate_btn = gr.Button("✨ Generate", variant="primary")
+
+            with gr.Column(scale=1):
+                output_image = gr.Image(
+                    label="Generated Flowchart",
+                    type="filepath",
+                    interactive=False,
+                    value=create_placeholder_image(),
+                    height=600,
+                    show_label=False
+                )
+                download_btn = gr.DownloadButton(
+                    "⬇️ Download",
+                    variant="primary",
+                    visible=False, # Initially hidden
+                )
+
+    # --- Event Listeners ---
+    generate_btn.click(
+        fn=handle_generate_click,
+        inputs=[prompt_input],
+        outputs=[output_image, download_btn]
+    )
+    
+    # Pre-load the default example when the app starts
+    demo.load(
+        fn=handle_generate_click,
+        inputs=[prompt_input],
+        outputs=[output_image, download_btn]
+    )
+
+if __name__ == "__main__":
+    demo.launch()