Update app.py

app.py

@@ -54,16 +54,72 @@ 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",
+            "description": "Provides a detailed recipe strictly adhering to the user input/specifications, especially ingredient exclusions and the recipe difficulty",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "name": {
+                        "type": "string",
+                        "description": "A creative name for the recipe"
+                    },
+                    "description": {
+                        "type": "string",
+                        "description": "a brief one-sentence description of the provided recipe"
+                    },
+                    "ingredients": {
+                        "type": "array",
+                        "items": {
+                            "type": "object",
+                            "properties": {
+                                "name": {
+                                    "type": "string",
+                                    "description": "Quantity and name of the ingredient"
+                                }
+                            }
+                        }
+                    },
+                    "instructions": {
+                        "type": "array",
+                        "items": {
+                            "type": "object",
+                            "properties": {
+                                "step_number": {
+                                    "type": "number",
+                                    "description": "The sequence number of this step"
+                                },
+                                "instruction": {
+                                    "type": "string",
+                                    "description": "Detailed description of what to do in this step"
+                                }
+                            }
+                        }
+                    }
+                },
+                "required": [
+                    "name",
+                    "description",
+                    "ingredients",
+                    "instructions"
+                    ],
+            },
+          }
+        ]
+
+        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_new_tokens=10000,
+            max_tokens=10000,
             top_p=0.9,
-            repetition_penalty=1.0,
-            do_sample=True,
+            frequency_penalty=1.0,
+            # do_sample=True,
         )
 
-        response = client.text_generation(prompt, **generate_kwargs)
+        response = client.chat_completion(prompt, **generate_kwargs)
         st.session_state.recipe = response
         st.session_state.recipe_saved = False
         
@@ -147,17 +203,17 @@ with button_cols_submit[2]:
 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"
-    # recipe_md += '\n## Instructions:\n'
-    # for instruction in recipe['instructions']:
-    #     recipe_md += f"{instruction['step_number']}. {instruction['instruction']} \n"
+    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"
+    recipe_md += '\n## Instructions:\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(st.session_state.recipe)
+    st.markdown(recipe_md)
     st.write("")