Update app.py

app.py

@@ -3,6 +3,8 @@ from sentence_transformers import SentenceTransformer, util
 import openai
 import os
 import random
+import transformers
+from transformers import pipeline
 
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
@@ -94,7 +96,17 @@ def generate_response(user_query, relevant_segment):
         ai_int=random.randint(0,10)
         ai_helpers=["https://chatgpt.com/ - An AI chatbot","https://www.grammarly.com/ - Help with grammar and writing!","https://www.any.do/ - Creates a to do list to help you get your tasks completed!","https://scheduler.ai/- AI optimizes your schedule and works around pre-scheduled deadlines","ChatGPT Data Analyst - Helps you visualize and analize your data","ChatGPT Logo creator - Helps to create professional logos for companies or brands","ScholarGPT - Enhances your reaserch capabilities","ChatGPT's Math solver","Tutor Me by Khan Academy","Travel Guide by capchair - helps find destinations, plan trips, and manage budgets"]
         output_text=output_text+"\n\n Here is a helpful chatbot tool for you!: "+ ai_helpers[ai_int-1]
-        return output_text
+        #return output_text
+
+        # 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 AI: {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, output_text
         
     except Exception as e:
         print(f"Error in generating response: {e}")