Update app.py

app.py

@@ -1,5 +1,4 @@
 import streamlit as st
-from PIL import Image
 from ibm_watsonx_ai import APIClient
 from ibm_watsonx_ai import Credentials
 from ibm_watsonx_ai.foundation_models import ModelInference
@@ -61,12 +60,27 @@ if api_key and project_id:
             Dimensions: {dimensions}\n
             Constraints: {constraints}\n
             Budget: {budget} USD\n
-            Provide detailed design concepts and explain their features."""
+            Provide detailed design concepts, explaining how they meet the constraints and budget. Also, suggest alternatives if the current design exceeds the budget or constraints."""
             
             try:
                 response = model.generate_text(prompt=prompt, params=parameters)
                 st.success("Generated Design Concepts:")
                 st.write(response)
+                
+                # Follow-up options for further queries
+                st.subheader("Refine Your Design")
+                if st.button("Ask for a cheaper variant"):
+                    follow_up_prompt = prompt + "\nPlease suggest a cheaper variant."
+                    follow_up_response = model.generate_text(prompt=follow_up_prompt, params=parameters)
+                    st.info("Cheaper Variant:")
+                    st.write(follow_up_response)
+                
+                if st.button("Explore alternative materials"):
+                    follow_up_prompt = prompt + "\nPlease explore alternative materials that might better fit the design constraints."
+                    follow_up_response = model.generate_text(prompt=follow_up_prompt, params=parameters)
+                    st.info("Alternative Materials:")
+                    st.write(follow_up_response)
+
             except Exception as e:
                 st.error(f"An error occurred: {e}")