Update app.py

app.py

@@ -1,68 +1,73 @@
-
 import streamlit as st
 from PIL import Image
-import datetime
-import time
 from ibm_watsonx_ai import APIClient
 from ibm_watsonx_ai import Credentials
-import os
-from ibm_watsonx_ai.foundation_models.utils.enums import ModelTypes
 from ibm_watsonx_ai.foundation_models import ModelInference
+from ibm_watsonx_ai.foundation_models.utils.enums import ModelTypes, DecodingMethods
 from ibm_watsonx_ai.metanames import GenTextParamsMetaNames as GenParams
-from ibm_watsonx_ai.foundation_models.utils.enums import DecodingMethods
-import requests
-import json
-
-st.title("August 2024 IBM Hackathon")
-st.header("Sinple app to find synonyms")
-
-st.sidebar.header("About")
-st.sidebar.text("Developed by Tony Pearson")
-
-project_id = os.environ['WATSONX_PROJECT_ID']
+import os
 
-credentials = Credentials(
-    url = "https://us-south.ml.cloud.ibm.com",
-    api_key = os.environ['WATSONX_API_KEY']
-)
+# Setting up the page layout
+st.set_page_config(page_title="AI Product Design & Development", layout="wide")
 
-client = APIClient(credentials)
-client.set.default_project(project_id)
+# Sidebar - User inputs for Product Specifications
+st.sidebar.title("Product Specifications")
+product_name = st.sidebar.text_input("Product Name", "Example Product")
+material = st.sidebar.selectbox("Material", ["Plastic", "Metal", "Wood", "Composite"])
+dimensions = st.sidebar.text_input("Dimensions (L x W x H in cm)", "10 x 5 x 3")
+constraints = st.sidebar.text_area("Design Constraints", "E.g., Must be lightweight, eco-friendly")
+budget = st.sidebar.number_input("Budget ($)", min_value=0, value=1000)
 
-parameters = {
-    GenParams.DECODING_METHOD: DecodingMethods.GREEDY,
-    GenParams.MIN_NEW_TOKENS: 1,
-    GenParams.MAX_NEW_TOKENS: 50,
-    GenParams.STOP_SEQUENCES: ["\n"]
-}
+st.sidebar.subheader("Project Info")
+st.sidebar.text("AI-Powered Product Design")
 
-model_id = ModelTypes.GRANITE_13B_CHAT_V2
+# Main app title and description
+st.title("AI Product Design & Development Tool")
+st.markdown("""
+Welcome to the AI-powered product design and development tool. This app leverages generative AI to accelerate the design process, optimize products for manufacturing, and simulate product performance.
+""")
 
-model = ModelInference(
-    model_id=model_id,
-    params=parameters,
-    credentials=credentials,
-    project_id = project_id
-    )
-    
-input_word = st.text_input("Enter your input word", "")
-    
-prompt_txt = """Generate a list of words similar to the input word."
+# Tabs for different sections of the app
+tabs = st.tabs(["Design Generation", "Simulation", "Optimization"])
 
-Input: wordy
-Output: verbose, loquacious, talkative
+# IBM WatsonX API Setup
+project_id = os.getenv('WATSONX_PROJECT_ID')
+api_key = os.getenv('WATSONX_API_KEY')
 
-Input: """
+if api_key and project_id:
+    credentials = Credentials(url="https://us-south.ml.cloud.ibm.com", api_key=api_key)
+    client = APIClient(credentials)
+    client.set.default_project(project_id)
 
-prompt_input = prompt_txt + input_word + "\nOutput:"
+    parameters = {
+        GenParams.DECODING_METHOD: DecodingMethods.GREEDY,
+        GenParams.MIN_NEW_TOKENS: 50,
+        GenParams.MAX_NEW_TOKENS: 200,
+        GenParams.STOP_SEQUENCES: ["\n"]
+    }
 
-st.code(prompt_input)
+    model_id = ModelTypes.GRANITE_13B_CHAT_V2
+    model = ModelInference(model_id=model_id, params=parameters, credentials=credentials, project_id=project_id)
 
-gen_parms_override = None
+    # Design Generation Tab
+    with tabs[0]:
+        st.header("Generate Product Designs")
+        st.write("Input your product specifications in the sidebar and click below to generate design concepts.")
+        
+        if st.button("Generate Design Concepts"):
+            prompt = f"""You are an AI specialized in product design. Generate creative product design concepts based on the following details:\n
+            Product Name: {product_name}\n
+            Material: {material}\n
+            Dimensions: {dimensions}\n
+            Constraints: {constraints}\n
+            Budget: {budget} USD\n
+            Provide detailed design concepts and explain their features."""
+            
+            try:
+                response = model.generate_text(prompt=prompt, params=parameters)
+                st.success("Generated Design Concepts:")
+                st.write(response)
+            except Exception as e:
+                st.error(f"An error occurred: {e}")
 
-# GENERATE
-   
-if st.button("Generate list of synonyms"):
-    generated_text_response = model.generate_text(prompt=prompt_input, params=parameters)
-    st.write("Output from generate_text() method:")
-    st.success(generated_text_response)
+# Simulation and Optimization tabs will be expanded in future steps.