Update app.py

app.py

@@ -1,8 +1,7 @@
 import gradio as gr
 from sentence_transformers import SentenceTransformer, util
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+import openai  
 import os
-import torch
 
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
@@ -10,9 +9,7 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
 filename = "output_topic_details.txt"  # Path to the file storing destress-specific details
 retrieval_model_name = 'output/sentence-transformer-finetuned/'
 
-# Load GPT-2 model and tokenizer
-tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
-model = GPT2LMHeadModel.from_pretrained("gpt2")
+#openai.api_key = os.environ["OPENAI_API_KEY"]
 
 system_message = "You are a comfort chatbot specialized in providing information on therapy, destressing activities, and student opportunities."
 messages = [{"role": "system", "content": system_message}]
@@ -75,7 +72,13 @@ def generate_response(user_query, relevant_segment):
         output_text = tokenizer.decode(output[0], skip_special_tokens=True)
 
         # Append assistant's message to messages list for context
-        messages.append({"role": "assistant", "content": output_text})
+        messages.append({"role": "user", "content": user_message})
+        
+        response = openai.ChatCompletion.create(
+            model="gpt-4o",
+            messages=messages,
+            max_tokens=4000,
+            temperature=0.5,
         return output_text
 
     except Exception as e: