Update app.py

app.py

@@ -54,6 +54,7 @@ def create_detailed_prompt(user_direction, exclusions, serving_size, difficulty)
 def generate_recipe(user_inputs):
     with st.spinner('Building the perfect recipe...'):
         prompt = create_detailed_prompt(user_inputs['user_direction'], user_inputs['exclusions'], user_inputs['serving_size'], user_inputs['difficulty'])
+
         functions = [
           {
             "name": "provide_recipe",
@@ -108,27 +109,26 @@ def generate_recipe(user_inputs):
           }
         ]
 
-        prompt += f"\nPlease format the output in JSON. The JSON should include fields for 'name', 'description', 'ingredients', and 'instructions', with each field structured as described below.\n\n{functions}"
-
-
         generate_kwargs = dict(
             temperature=0.9,
-            max_tokens=10000,
+            max_new_tokens=10000,
             top_p=0.9,
-            frequency_penalty=1.0,
-            # do_sample=True,
+            repetition_penalty=1.0,
+            do_sample=True,
         )
 
-        response = client.chat_completion(prompt, **generate_kwargs)
+        prompt += f"\nPlease format the output in JSON. The JSON should include fields for 'name', 'description', 'ingredients', and 'instructions', with each field structured as described below.\n\n{functions}"
+
+        response = client.text_generation(prompt, **generate_kwargs)
         st.session_state.recipe = response
         st.session_state.recipe_saved = False
-        
+
 def clear_inputs():
     st.session_state.user_direction = None
     st.session_state.exclusions = None
     st.session_state.serving_size = 2
     st.session_state.selected_difficulty = "Quick & Easy"
-    
+
 st.title("Let's get cooking")
 st.session_state.user_direction = st.text_area(
     "What do you want to cook? Describe anything - a dish, cuisine, event, or vibe.",
@@ -159,12 +159,12 @@ difficulty_dictionary = {
 st.session_state.selected_difficulty = st.radio(
     "Choose a difficulty level for your recipe.",
     [
-        list(difficulty_dictionary.keys())[0], 
-        list(difficulty_dictionary.keys())[1], 
+        list(difficulty_dictionary.keys())[0],
+        list(difficulty_dictionary.keys())[1],
         list(difficulty_dictionary.keys())[2]
     ],
     captions = [
-        difficulty_dictionary["Quick & Easy"]["description"], 
+        difficulty_dictionary["Quick & Easy"]["description"],
         difficulty_dictionary["Intermediate"]["description"],
         difficulty_dictionary["Professional"]["description"]
     ],
@@ -181,10 +181,10 @@ fancy_exclusions = ""
 
 if st.session_state.selected_difficulty == "Professional":
     exclude_fancy = st.checkbox(
-        "Exclude cliche professional ingredients? (gold leaf, truffle, edible flowers, microgreens)", 
+        "Exclude cliche professional ingredients? (gold leaf, truffle, edible flowers, microgreens)",
         value=True)
     fancy_exclusions = "gold leaf, truffle, edible flowers, microgreens, gold dust"
-      
+
 
 user_inputs = {
     "user_direction" : st.session_state.user_direction,
@@ -199,21 +199,23 @@ with button_cols_submit[1]:
     st.button(label='Reset', on_click=clear_inputs, type="secondary", use_container_width=True)
 with button_cols_submit[2]:
     st.empty()
-    
+
 if st.session_state.recipe is not None:
     st.divider()
-    print(st.session_state.recipe)
-    recipe = json.loads(st.session_state.recipe)
-    recipe_md = ''
-    recipe_md += f'# {recipe["name"]} \n\n'
-    recipe_md += f'{recipe["description"]} \n\n'
-    recipe_md += '## Ingredients: \n'
-    for ingredient in recipe['ingredients']:
-        recipe_md += f"- {ingredient['name']} \n"
+    print(st.session_state.recipe,"\n\n\n\n")
+    recipe = ast.literal_eval(st.session_state.recipe)
+    recipe_md = f'# {recipe["name"]}\n\n'
+    recipe_md += f'{recipe["description"]}\n\n'
+    recipe_md += '## Ingredients:\n'
+    # print("ingredients : ",recipe[recipe.index("ingredients"):recipe.index('instructions')][len("ingredients")-1:])
+    for ingredient in recipe["ingredients"]:
+        recipe_md += f"- {ingredient['name']}\n"
     recipe_md += '\n## Instructions:\n'
-    for instruction in recipe['instructions']:
-        recipe_md += f"{instruction['step_number']}. {instruction['instruction']} \n"
+    for instruction in recipe["instructions"]:
+        recipe_md += f"{instruction['step_number']}. {instruction['instruction']}\n"
+
     # recipe['md'] = recipe_md
     # recipe['timestamp'] = str(datetime.now())
-    st.markdown(recipe_md)
-    st.write("")
+
+    st.markdown((recipe_md))
+    st.write("")