Update app.py

app.py

@@ -208,7 +208,7 @@ def function3(input_text,one_shot_example,n,llm):
                   )
               ]
           )
-        return generated_answer.content
+        return generated_answer.content,results
       else:
         combined_input = k+f"\n\n Here is a sample question answer pair for reference :\n\n {one_shot_example} "
         generated_answer = model(
@@ -218,7 +218,7 @@ def function3(input_text,one_shot_example,n,llm):
                   )
               ]
           )
-        return generated_answer.content
+        return generated_answer.content,results
   else:
       model,client=llm_route(llm)
       if len(one_shot_example)==0:
@@ -232,7 +232,7 @@ def function3(input_text,one_shot_example,n,llm):
                 stop=None,  # Optional: specify a stop sequence
                 temperature=0.7  # Adjust the creativity of the response
             )
-        return completion.choices[0].text.strip()
+        return completion.choices[0].text.strip(),results
       else:
         combined_input = k+f"\n\n Here is a sample question answer pair for reference :\n\n {one_shot_example} "
         completion = client.completions.create(
@@ -274,7 +274,7 @@ def function4(input_text, one_shot_example, n,llm):
                       )
                   ]
               )
-            return generated_answer.content
+            return generated_answer.content,results
         else:
             
             combined_input = f"please provide comprehensive and well-researched responses to the following question. Ensure that the information is up-to-date and includes relevant scientific insights and data  \n\n context:{k} \n\n,Below is an example question-answer pair for reference\n\n {one_shot_example} \n\n  Now answer this question \n\n question :{input_text}"
@@ -285,7 +285,7 @@ def function4(input_text, one_shot_example, n,llm):
                       )
                   ]
               )
-            return generated_answer.content
+            return generated_answer.content,results
     else:
         model,client=llm_route(llm)
         if len(one_shot_example)==0:
@@ -298,7 +298,7 @@ def function4(input_text, one_shot_example, n,llm):
                 stop=None,  # Optional: specify a stop sequence
                 temperature=0.7  # Adjust the creativity of the response
             )
-            return completion.choices[0].text.strip()
+            return completion.choices[0].text.strip(),results
         else:
             completion = client.completions.create(
                 model=model,