Update app.py

app.py

@@ -122,7 +122,7 @@ tools = [
 	)
 ]
 
-prefix = "Give a detailed answer to the question"
+prefix = "Give an answer to the question"
 suffix = """Give answer intended for knowledgeable doctors
 
 Question: {input}
@@ -147,29 +147,59 @@ def get_answer(query_string):
 	return result
 
 
-def get_answer2(list_thing):
+def get_answer2(list_thing, answer_type):
 	responses = []
-	for question in list_thing:
+	if answer_type == "Detailed Answer":
+		for question in list_thing:
+			answer = get_answer(question)
+			response = f"{question}\n{answer}"
+			responses.append(response)
+			
+		return "\n\n".join(responses)
+	elif answer_type == "Consolidated answer":
+		for question in list_thing:
+			answer = get_answer(question)
+			response = f"{question}\n{answer}"
+			responses.append(response)
+			
+		return "\n\n".join(responses)
+	
+	else:
+		question = list_thing[0]
 		answer = get_answer(question)
 		response = f"{question}\n{answer}"
-		responses.append(response)
-		
-	return "\n\n".join(responses)
+		return response
+
+
 
 
 	
 
 
 
-def qa_app(query):
+def qa_app(query, answer_type):
 	question_list = generate_variations(query)
-	big_answer = get_answer2(question_list)
-	#final_answer = consolidated_answer(big_answer, query)
-	return big_answer
-
+	big_answer = get_answer2(question_list, answer_type)
+	if answer_type == "Consolidated answer":
+		final_answer = consolidated_answer(big_answer, query)
+		return final_answer
+	else:
+		return big_answer
+	
+	
+inputs = [
+	gr.inputs.Textbox(label="Enter your question:"),
+	gr.inputs.Dropdown(choices=["Quick Answer", "Detailed Answer", "Consolidated answer"], label="Choose answer type:")
+]
 
-inputs = gr.inputs.Textbox(label="Enter your question:")
 output = gr.outputs.Textbox(label="Answer:")
-iface = gr.Interface(fn=qa_app, inputs=inputs, outputs=output, title="Endo AI : Endocrine answering app by Dr. Om J Lakhani")
+
+iface = gr.Interface(
+	fn=qa_app,
+	inputs=inputs,
+	outputs=output,
+	title="Endo AI : Endocrine answering app by Dr. Om J Lakhani"
+)
 
 iface.launch()
+