Update app.py

app.py

@@ -83,6 +83,32 @@ def retrieve_info(query):
         formatted_results.append(formatted_result)
         
     return {"results": formatted_results}
+        # Format the reranked results and append to user prompt
+    user_prompt = f"User: {query}\n"
+    for idx, r in enumerate(reranked_results):
+        user_prompt += f"Document {idx + 1}: {r.document['text']}\nRelevance Score: {r.relevance_score:.2f}\n\n"
+
+    # Final API call to OpenAI
+    final_response = client.chat.completions.create(
+        model="gpt-4-1106-preview",
+        messages=[
+            {
+                "role": "system",
+                "content": "You are a redditor. Assess, rephrase, and explain the following. Provide long answers. Use the same words and language you receive."
+            },
+            {
+                "role": "user",
+                "content": user_prompt
+            }
+        ],
+        temperature=1.63,
+        max_tokens=2240,
+        top_p=1,
+        frequency_penalty=1.73,
+        presence_penalty=1.76
+    )
+
+    return final_response.choices[0].text
 
 def combined_interface(query, file, collection_name):
     if query: