Update app.py

app.py

@@ -49,10 +49,27 @@ def summarize_text(text: str):
         presence_penalty=0.0
     )
     if response.choices:
+        # Extract the text from the response
         summary = response.choices[0].text.strip()
-        return GenerateResponse(text=summary)
+        
+        # Use regular expressions to find the keywords/metrics and summary sections
+        keywords_match = re.search(r"Keywords/Metrics:(.*?)(\n|$)", summary, re.S)
+        summary_match = re.search(r"Summary:(.*?)(\n|$)", summary, re.S)
+        
+        # Extract the content if matches are found
+        keywords = keywords_match.group(1).strip() if keywords_match else "No keywords found"
+        conversation_summary = summary_match.group(1).strip() if summary_match else "No summary found"
+        
+        # Return a clean version with just the keywords and summary
+        return GenerateResponse(text=f"Keywords/Metrics: {keywords}\nSummary: {conversation_summary}")
+    
     else:
         return "No response from the API."
+    # if response.choices:
+    #     summary = response.choices[0].text.strip()
+    #     return GenerateResponse(text=summary)
+    # else:
+    #     return "No response from the API."
 
 
 # QA Automation function