Update app.py

app.py

@@ -238,6 +238,32 @@ def convert_to_markdown(vectara_response_json):
         return "No data found in the response."
 # Main function to handle the Gradio interface logic
 
+def process_summary_with_openai(summary):
+    """
+    This function takes a summary text as input and processes it with OpenAI's GPT model.
+    """
+    try:
+        # Ensure that the OpenAI client is properly initialized
+        client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'))
+        
+        # Create the prompt for OpenAI's completion
+        prompt = "Please summarize the following text in two lines and make it more understandable:"
+        
+        # Call the OpenAI API with the prompt and the summary
+        completion = client.chat.completions.create(
+            model="gpt-3.5-turbo",  # Make sure to use the correct model name
+            messages=[
+                {"role": "system", "content": prompt},
+                {"role": "user", "content": summary}
+            ]
+        )
+        
+        # Extract the content from the completion
+        final_summary = completion.choices[0].message.content
+        return final_summary
+    except Exception as e:
+        return str(e)
+        
 
 def process_and_query(text=None):
     try: