Update app.py

app.py

@@ -3,6 +3,11 @@ from langchain_community.document_loaders import JSONLoader
 from langchain_community.vectorstores import Qdrant
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from sentence_transformers.cross_encoder import CrossEncoder
+from groq import Groq
+
+client = Groq(
+    api_key=os.environ.get("GROQ_API"),
+)
 
 # loading data
 json_path = "format_food.json"
@@ -64,14 +69,24 @@ def run_query(query):
     instructions = format_to_markdown(answer[0].metadata['instructions'])
     recipe = f"# Cooking time:\n{answer[0].metadata['time']}\n\n# Recipe:\n{instructions}"
     print("Returning query")
-    return title_and_description, recipe
+    chat_completion = client.chat.completions.create(
+        messages=[
+           {
+                "role": "user",
+                "content": f"please write a more detailed recipe for the following recipe:\n{recipe}\n\n please return it in the same format.",
+            }
+        ],
+        model="Llama3-70b-8192",
+    )
+    return title_and_description, recipe, chat_completion[0].text
 
 with gr.Blocks() as demo:
     gr.Markdown("Start typing below and then click **Run** to see the output.")
     inp = gr.Textbox(placeholder="What sort of meal are you after?")
     title_output = gr.Markdown(label="Title and description")
     instructions_output = gr.Markdown(label="Recipe")
+    updated_recipe = gr.Markdown(label="Updated Recipe")
     btn = gr.Button("Run")
-    btn.click(fn=run_query, inputs=inp, outputs=[title_output, instructions_output])
+    btn.click(fn=run_query, inputs=inp, outputs=[title_output, instructions_output, updated_recipe])
 
 demo.launch()