Update app.py

app.py

@@ -1,3 +1,4 @@
+# imports
 import streamlit as st
 import numpy as np
 import pandas as pd
@@ -6,9 +7,14 @@ import json
 import openai
 openai.api_key = st.secrets["open_ai_key"]
 
+# state management
+if 'gpt_response' not in st.session_state:
+    st.session_state.gpt_response = None
+
+# app
 user_direction = st.text_area(
     "Let's get cooking! What do you feel like making?",
-    placeholder="quick snack, asian style bowl with either noodles or rice, something italian, ...",
+    placeholder="quick snack, asian style bowl with either noodles or rice, something italian",
     )
 
 serving_size = st.number_input(
@@ -28,16 +34,21 @@ difficulty = st.radio(
         "Complex recipes that demand a high level of skill and precision. Suited for seasoned cooks aspiring to professional-level sophistication and creativity."
     ])
 
+exclusions = st.text_area(
+    "Any ingredients you want to exclude?",
+    placeholder="gluten, dairy, nuts, cilantro",
+    )
+
 user_inputs = {
     "user_direction" : user_direction,
-    "exclusions": "gluten, dairy",
-    "serving_size": 4,
-    "difficulty": "easy"
+    "exclusions": exclusions,
+    "serving_size": serving_size,
+    "difficulty": difficulty
 }
 
-# define GPT agents
-def generate_recipe(text):
-    context = """Provide me a recipe based on the below user input. The recipe output should contain the following components:
+    
+def generate_recipe(user_inputs):
+    context = """Provide me a recipe based on the user input. The recipe output should contain the following components:
     - recipe name
     - serving size
     - time to make
@@ -52,14 +63,17 @@ def generate_recipe(text):
     """
     messages = [
         {"role": "system", "content": context},
-        {"role": "user", "content": text}
+        {"role": "user", "content": str(user_inputs)}
         ]
-    response = openai.ChatCompletion.create(
+    
+    st.session_state.gpt_response = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
         messages=messages,
         temperature=0.2
     )
-    response_message = response["choices"][0]["message"]
-    return json.loads(response_message["content"])
 
-#generate_recipe(str(user_inputs))
+
+st.button(label='Submit', on_click=get_response, kwargs=dict(user_inputs=user_inputs))
+
+if st.session_state.gpt_response is not None:
+    st.json(json.loads(st.session_state.gpt_response['choices'][0]['message']['content']))