Update app.py

app.py

@@ -101,11 +101,11 @@ def chatbot_response(user_query, top_k=3, num_candidates=3):
     best_response = grade_responses(speculative_responses, user_query)
     return best_response
 
-def chatbot(query):
+def chatbot(query, num_candidates):
     if not query.strip():
         return "Please ask a question about hadiths."
     try:
-        answer = chatbot_response(query)
+        answer = chatbot_response(query, num_candidates)
         if "don't know" in answer.lower() or "not sure" in answer.lower():
             return "Sorry. I don't have information about the hadiths related. It might be a dhoif, or maudhu, or I just don't have the knowledge."
         else:
@@ -117,7 +117,9 @@ def chatbot(query):
 if __name__ == "__main__":
     iface = gr.Interface(
         fn=chatbot,
-        inputs=gr.Textbox(lines=7, placeholder="Ask me a question about hadiths...", label="Question"),
+        inputs=[
+            gr.Textbox(lines=2, placeholder="Enter your question here..."),
+            gr.Slider(minimum=1, maximum=10, value=3, step=1, label="Number of Hadiths as References")],
         outputs=gr.Textbox(label="Answer"),
         title="Hadith QA Chatbot",
         description="Ask questions related to Hadiths."