Update app.py

app.py

@@ -64,6 +64,7 @@ def generate_response(user_query, relevant_segment):
     """
     Generate a response emphasizing the bot's capability to provide information related to composting food.
     """
+    """
     try:
         system_message = "You are a chatbot specialized in providing information about food composting tips, tricks, and basics."
         user_message = f"Here's the information on composting: {relevant_segment}"
@@ -84,6 +85,16 @@ def generate_response(user_query, relevant_segment):
     except Exception as e:
         print(f"Error in generating response: {e}")
         return f"Error in generating response: {e}"
+    """
+    # Create pipeline for text generation with confidence scores
+    generator = pipeline("text-davinci-003", device=0)  # Adjust device if needed
+
+    # Generate response and get confidence score
+    response = generator(query=f"Here's the information on chess: {relevant_segment} {user_query}", max_length=150, temperature=0.2, top_p=1)
+    generated_text = response[0]['generated_text'].strip()
+    confidence_score = response[0]['score']
+
+    return generated_text, confidence_score
 
 def query_model(question):
     """