File size: 5,176 Bytes
ed7adeb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 |
from fastapi import FastAPI, File, UploadFile
from fastapi.responses import FileResponse
from groq import Groq
import os
import base64
from io import BytesIO
from PIL import Image
from mdextractor import extract_md_blocks
from pydantic import BaseModel
from processpiper.text2diagram import render
from dotenv import load_dotenv
load_dotenv()
app = FastAPI()
PIPERFLOW_SYNTAX_DOC=""" Generate BPMN diagram using English like PiperFlow syntax
To create a process map using PirperFlow, you need to define the diagram using a specific syntax. Here is an example:
title: Sample Test Process
colourtheme: GREENTURTLE
lane: End User
(start) as start
[Enter Keyword] as enter_keyword
(end) as end
pool: System Search
lane: Database System
[Login] as login
[Search Records] as search_records
<Result Found?> as result_found
[Display Result] as display_result
[Logout] as logout
lane: Log System
[Log Error] as log_error
start->login->enter_keyword->search_records->result_found->display_result->logout->end
result_found->log_error->display_result
footer: Generated by ProcessPiper
Import the render function from processpiper.text2diagram
Call the render function with the input syntax and the output file name.
Syntax
Diagram Configurations
The PiperFlow syntax for defining a process map is as follows:
title: The title of the diagram.
footer: The footer text to display on the diagram.
width : (Optional) Specify the width of the diagram.
colourtheme: The colour theme to use
Lane & Pool Configurations
lane: The name of the lane.
pool: The name of the pool.
Element/Shape Configurations
To add elements to the lane, use one of the following tags. You place your element description within the tag:
Use ( and ) to create event element
use (start) to create a start event
use (end) to create an end event
use (@timer and ) to create a timer event. Example (@timer Trigger every 1 hour) as timer_event
use (@intermediate and ) to create an intermediate event. Example (@intermediate Message Received) as intermediate_event
use (@message and ) to create a message event
use (@signal and ) to create a signal event
use (@conditional and ) to create a conditional event
use (@link and ) to create a link event
Use [ and ] to create an activity. By default, the activity type is TASK. Example [Place Order] as place_order
use [@subprocess] to create a subprocess. Example `[@subprocess Get Approval] as get_approval``
Use < and > to create a gateway. By default, the gateway type is EXCLUSIVE. Example <Result Found?> as result_found
Use <@parallel and > to create a parallel gateway. Example <@parallel Span Out> as span_out
Use <@inclusive and > to create an inclusive gateway. Example <@inclusive Condition Met?> as condition_met
Use <@event and > to create an event gateway
Connection Configurations
To connect two elements, use ->. You can chain multiple connections using ->:
Example:
login->enter_keyword
start->login->enter_keyword->search_records->result_found->display_result->logout->end
To add label to the connection, add ":" when connecting elements. ❗ NOTE: This is a breaking change in v0.6. Versions prior to 0.6 use start-"Enter Credentials->login syntax. See this page for more information.
Example:
start->login: Enter credentials
To specify the connection point manually, add connection side. See How to select sides for more information.
Example:
start-(bottom, top)->login
start-(bottom, top)->login: Enter credentials
Indentation is not required. However, it is recommended to use indentation to make the diagram easier to read.
currently available color themes are
Default
GREYWOOF
BLUEMOUNTAIN
ORANGEPEEL
GREENTURTLE
SUNFLOWER
PURPLERAIN
RUBYRED
TEALWATERS
SEAFOAMS
"""
GROQ_API_KEY = os.getenv("GROQ_API_KEY")
class Prompt(BaseModel):
prompt: str
def generate_diagram(input_syntax):
_, generated_image = render(input_syntax)
# Convert the image to a base64 encoded string
buffered = BytesIO()
generated_image.save(buffered, format="PNG")
img_str = base64.b64encode(buffered.getvalue())
return img_str.decode("utf-8")
@app.get("/")
def read_root():
return FileResponse("index.html")
@app.post("/generate/")
async def generate_bpmn(prompt:Prompt):
client = Groq(api_key=GROQ_API_KEY)
completion = client.chat.completions.create(
model="llama3-70b-8192",
messages=[
{
"role": "system",
"content": "you are business process flow generator using the following piperflow text\n\n" + PIPERFLOW_SYNTAX_DOC
},
{
"role": "user",
"content": "generate the piperflow text for the below scenario\n\n" + prompt.prompt
}
],
temperature=0.2,
max_tokens=1024,
top_p=1,
stream=False,
stop=None,
)
print(completion.choices[0].message.content or "", end="")
piperFlowText=extract_md_blocks(completion.choices[0].message.content)[0]
print(piperFlowText)
return {"pipeFlowImage":generate_diagram(piperFlowText),"pipeFlowText":piperFlowText}
|