Update app.py

app.py

@@ -5,7 +5,6 @@ import openai
 from sentence_transformers import SentenceTransformer
 from nltk.tokenize import sent_tokenize
 import nltk
-import torch
 
 # Download the required NLTK data
 nltk.download('punkt')
@@ -91,20 +90,22 @@ def rag_response(query, k=5, max_tokens=150):
         augmented_input = query + "\n" + "\n".join(relevant_chunks)
         
         # Generate response using OpenAI API
-        response = openai.Completion.create(
+        response = openai.ChatCompletion.create(
             model="gpt-3.5-turbo",
-            prompt=augmented_input,
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "user", "content": augmented_input}
+            ],
             max_tokens=max_tokens,
             temperature=0.7
         )
-        generated_text = response.choices[0].text.strip()
+        generated_text = response.choices[0].message['content'].strip()
         return generated_text, distances, indices
     except Exception as e:
         return f"An error occurred: {e}", [], []
 
 # Gradio Interface
 def format_output(response, distances, indices):
-    # Format output to include distances and indices
     formatted_response = f"Response: {response}\n\nDistances: {distances}\n\nIndices: {indices}"
     return formatted_response