Update app.py

app.py

@@ -54,8 +54,14 @@ def Get_Inference(text_input):
     # Generate the prompt with the text input
     formatted_prompt = prompt.format(format_instructions=format_instructions, text=text_input)
     # Send the prompt to the LLM and get the response
-    response = llm.invoke(formatted_prompt)  # Pass a list with a single prompt string
-    return str(response)
+    response = llm.invoke(formatted_prompt)  # Pass a list with a single prompt string    
+    data = json.loads(str(response.content).replace('\n','').replace('\t','').replace('json','').replace('`',''))
+    # Access individual values
+    order_placed = data["Order Placed"]
+    reason = data["Reason"]
+    customer_feedback = data["Customer Feedback"]
+    demands = data["Demands"]
+    return order_placed,reason,customer_feedback,demands
 
 
         
@@ -81,7 +87,10 @@ if sentence:
     results = Get_Inference(sentence)
     if len(results) > 0:
         st.write("Customer/Sales Inference:")
-        st.write(results)
+        st.write('Order Placed : ',results[0])
+        st.write('Reason       : ',results[1])
+        st.write('Feedback     : ',results[2])
+        st.write('Demand       : ',results[3])
     else:
         st.write("No Inference Available.")