Update app.py

app.py

@@ -38,16 +38,19 @@ def get_relevant_context(pdf_text, query, num_contexts=3):
     relevant_context = " ".join([pdf_text_chunks[i] for i in top_indices])
     return relevant_context
 
-# Function to generate a response from GPT-4o-mini model
+# Function to generate a response from GPT-4 chat model
 def generate_response(context, question):
-    prompt = f"The following is an excerpt from a research paper on GPT-4. Please answer the question based on this context:\n\nContext: {context}\n\nQuestion: {question}\nAnswer:"
-    response = openai.Completion.create(
-        model="gpt-4o-mini",  # Replace with the appropriate model identifier
-        prompt=prompt,
+    messages = [
+        {"role": "system", "content": "You are a helpful assistant expert on GPT-4."},
+        {"role": "user", "content": f"Context: {context}\nQuestion: {question}"}
+    ]
+    response = openai.ChatCompletion.create(
+        model="gpt-4o-mini",  # Use the GPT-4 chat model
+        messages=messages,
         max_tokens=200,
         temperature=0.7,
     )
-    return response.choices[0].text.strip()
+    return response['choices'][0]['message']['content'].strip()
 
 # Function to handle irrelevant questions
 def is_irrelevant_question(question):
@@ -85,7 +88,7 @@ def main():
                     # Get the most relevant context from the document
                     relevant_context = get_relevant_context(pdf_text, question)
                     
-                    # Generate the response from GPT-4o-mini
+                    # Generate the response from GPT-4 chat model
                     answer = generate_response(relevant_context, question)
                     
                     # Display the answer