t

app.py

@@ -56,7 +56,7 @@ I am building an online community to help people to find dates.
 {{"solution": "FindDates.com", "problem": "finding a date", "features": "online community to help people find dates", "target_customer": "people looking for a date", "fg_will_use": "True", "reason_to_use": "I am looking for an online community to help people find dates. FindDates.com meets my needs and I would use it to find my next great date.","fg_will_pay": "True", "reason_to_pay": "I would not pay for it as I am looking for an online community to help people find dates. But for products related to dating, paying for it would be a no-brainer.","fg_will_invest": "False", "reason_to_invest": "There are many online dating platforms already.","score": "40"}}
 """
 
-def extract_json(gen_text, n_shot_learning=0):
+def extract_json(gen_text, n_shot_learning=0, response):
     if(n_shot_learning == -1) :
         start_index = 0
     else :
@@ -69,7 +69,7 @@ def extract_json(gen_text, n_shot_learning=0):
     end_index = gen_text.find("}\n\n### ")
     if(end_index == -1) :
         print("Start***No end index!!!")
-        print(gen_text)
+        print(response)
         print("End***No end index!!!")
 
     end_index = end_index + 1
@@ -129,7 +129,7 @@ Give a score for the product. Format your response as a JSON object with \
 	print(total_prompt)
 	response = get_completion(total_prompt)
 	gen_text = response["choices"][0]["text"]
-	return json.dumps(extract_json(gen_text, -1))
+	return json.dumps(extract_json(gen_text, -1, response))
 
 #iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 #iface.launch()