Update app.py

app.py

@@ -3,8 +3,8 @@ from sentence_transformers import SentenceTransformer, util
 import openai
 import os
 import random
-import transformers
-from transformers import pipeline
+#import transformers
+#from transformers import pipeline
 
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
@@ -96,17 +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
+        #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']
+        #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
+        #return generated_text, confidence_score, output_text
         
     except Exception as e:
         print(f"Error in generating response: {e}")
@@ -155,9 +155,9 @@ with gr.Blocks(theme='JohnSmith9982/small_and_pretty') as demo:
             submit_button = gr.Button("Submit")
             submit_button.click(fn=query_model, inputs=question, outputs=answer)
 
-            def display_response(question):
-                response, confidence_score = query_model(question)
-                answer.value = f"Response: {response}\nConfidence Score: {confidence_score:.2f}"  
+            #def display_response(question):
+                #response, confidence_score = query_model(question)
+                #answer.value = f"Response: {response}\nConfidence Score: {confidence_score:.2f}"  
 
             #submit_button.click(fn=display_response, inputs=question, outputs=None)