update

app.py

@@ -34,7 +34,7 @@ A: Let’s think step by step.
 
 def text_generate(prompt, generated_txt): #, input_prompt_sql ): #, input_prompt_dalle2):
 
-  print(f"*****Inside poem_generate - Prompt is :{prompt}")
+  print(f"*****Inside text_generate - Prompt is :{prompt}")
   #if input_prompt_sql != '':
   #  prompt = input_prompt_sql #"Instruction: Given an input question, respond with syntactically correct PostgreSQL\nInput: " +input_prompt_sql + "\nPostgreSQL query: "
   #elif input_prompt_dalle2 !='':
@@ -56,10 +56,10 @@ def text_generate(prompt, generated_txt): #, input_prompt_sql ): #, input_prompt
   response = requests.post(API_URL, headers=headers, json=json_)
   print(f"Response  is : {response}")
   output = response.json()
-  print(f"output is : {output}") #{output}")
+  print(f"output is : {output}") 
   output_tmp = output[0]['generated_text']
   print(f"output_tmp is: {output_tmp}")
-  solution = output_tmp.split("\nQ:")[0]   #output[0]['generated_text'].split("Q:")[0] # +"."
+  solution = output_tmp.split("\nQ:")[0]   
   print(f"Final response after splits is: {solution}")
   if '\nOutput:' in solution:
     final_solution = solution.split("\nOutput:")[0] 
@@ -88,6 +88,17 @@ def text_generate(prompt, generated_txt): #, input_prompt_sql ): #, input_prompt
   new_prompt = final_solution[len(prompt):]
   print(f"new prompt for next cycle is : {new_prompt}")
   print(f"display_output for printing on screen is : {display_output}")
+  if len(new_prompt) == 0:
+    temp_text = display_output[::-1]
+    if temp_text[0] == '.':
+      first_period_loc = temp_text[1:].find('.')
+      new_prompt = display_output[:-first_period_loc]
+      print(f"Not sending blank as prompt so new prompt for next cycle is : {new_prompt}")
+    else:
+      first_period_loc = temp_text.find('.')
+      new_prompt = display_output[:-first_period_loc]
+      print(f"Not sending blank as prompt so new prompt for next cycle is : {new_prompt}")
+    
   return display_output, new_prompt  #generated_txt+prompt #final_solution