Update app.py

app.py

@@ -52,6 +52,35 @@ def find_relevant_segment(user_query, segments):
         print(f"Error in finding relevant segment: {e}")
         return ""
 
+def generate_response_with_context(user_query, relevant_segment):
+    """
+    Generate a response based on a user query and a relevant segment.
+    
+    Parameters:
+    - user_query (str): The user's query.
+    - relevant_segment (str): A relevant fact or detail.
+    
+    Returns:
+    - str: Formatted response incorporating the relevant segment.
+    """
+    try:
+        # Prepare the prompt incorporating the relevant segment
+        prompt = f"User: {user_query}\n\nAssistant: Here is some helpful information based on your topic: {relevant_segment}"
+        
+        # Calculate the maximum tokens allowed for the response
+        max_tokens = len(tokenizer(prompt)['input_ids']) + 100
+        
+        # Generate the response using the model
+        response = gpt_model(prompt, max_length=max_tokens, temperature=0.7)[0]['generated_text']
+        
+        # Clean up the response for better formatting and clarity
+        return clean_up_response(response, relevant_segment)
+    
+    except Exception as e:
+        print(f"Error generating response: {e}")
+        return "I'm sorry, but there was an error generating your response. Please try again."
+
+
 def generate_response(user_query, relevant_segment):
     try:
         user_message = f"Here's the information on your request: {relevant_segment}"