Create app.py

app.py

@@ -0,0 +1,38 @@
+import os
+import streamlit as st
+import requests
+from dotenv import load_dotenv
+
+# Load environment variables (for FLOWISE_API token)
+load_dotenv()
+
+# Define API settings
+API_URL = "https://nakheeltech.com:8030/api/v1/prediction/c1681ef1-8f47-4004-b4ab-594fbbd3eb3f"
+headers = {"Authorization": f"Bearer {os.getenv('FLOWISE_API')}"}
+
+# Function to send a query to the API
+def query(payload):
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.json()
+
+# Streamlit UI
+st.title("AWS Architecture Diagram Generator")
+
+# Get input from the user
+user_input = st.text_area("Describe your architecture:", 
+                          placeholder="Enter a description of your architecture here...")
+
+if st.button("Generate Diagram"):
+    if user_input:
+        # Send the user's input to the API
+        payload = {"question": user_input}
+        output = query(payload)
+        print(output)
+        
+        # Check and display the API response
+        if "diagram_url" in output:
+            st.image(output["diagram_url"], caption="Generated Architecture Diagram")
+        else:
+            st.write("Response:", output)
+    else:
+        st.warning("Please enter an architecture description.")