shukdevdatta123 commited on
Commit
42fedbe
·
verified ·
1 Parent(s): 5acc07a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -2,6 +2,7 @@ import streamlit as st
2
  import openai
3
  import urllib.parse
4
  import io
 
5
 
6
  # Function to call OpenAI GPT model for prompt processing
7
  def get_diagram_code(prompt, diagram_type, api_key):
@@ -41,6 +42,15 @@ def convert_svg_to_png(svg_file):
41
  def main():
42
  st.title("Generate Diagrams using GPT-4 and MermaidFlow")
43
 
 
 
 
 
 
 
 
 
 
44
  # User input for OpenAI API key
45
  api_key = st.text_input("Enter your OpenAI API Key:", type="password")
46
 
 
2
  import openai
3
  import urllib.parse
4
  import io
5
+ import cairosvg
6
 
7
  # Function to call OpenAI GPT model for prompt processing
8
  def get_diagram_code(prompt, diagram_type, api_key):
 
42
  def main():
43
  st.title("Generate Diagrams using GPT-4 and MermaidFlow")
44
 
45
+ # Display sample prompt
46
+ st.markdown("""
47
+ ### Sample Prompt For Users:
48
+
49
+ Create a UML diagram for a **Library Management System** that includes classes such as **Book** (with attributes like `bookID`, `title`, `author`, `publisher`, `genre`, `availabilityStatus` and methods like `checkAvailability()`, `updateAvailability()`), **Member** (with attributes such as `memberID`, `name`, `email`, `phoneNumber`, `membershipDate` and methods like `borrowBook()`, `returnBook()`, `reserveBook()`), **Staff** (with attributes such as `staffID`, `name`, `role`, `email`, `phoneNumber` and methods like `addBook()`, `removeBook()`, `updateBookInfo()`), **Transaction** (with attributes like `transactionID`, `transactionDate`, `dueDate`, `fineAmount` and methods like `calculateFine()`, `generateReceipt()`), and **Reservation** (with attributes like `reservationID`, `reservationDate`, `expirationDate` and methods like `cancelReservation()`, `checkReservationStatus()`).
50
+
51
+ Define the relationships where a **Member** can borrow many **Books**, a **Staff** can manage many **Books**, a **Transaction** is linked to one **Book** and one **Member**, and a **Reservation** is made by a **Member** for a **Book**. Ensure appropriate visibility markers for methods and attributes, and include necessary relationships such as associations, multiplicities, and inheritance. Optionally, consider adding sequence or activity diagrams for processes like book borrowing.
52
+ """)
53
+
54
  # User input for OpenAI API key
55
  api_key = st.text_input("Enter your OpenAI API Key:", type="password")
56