Update app.py

app.py

@@ -25,17 +25,31 @@ def generate_user_stories(text, prompt):
         return ""
 
 def format_as_user_stories(summary):
-    # Enhanced formatting logic to extract user stories
     user_stories = []
     lines = summary.split('. ')
     
     for line in lines:
-        # Improved pattern matching for user stories
         line = line.strip()
         if "as a" in line.lower() and "i want" in line.lower():
-            user_stories.append(line)
+            # Extract the parts of the user story
+            parts = line.split("so that")
+            if len(parts) == 2:
+                story = {
+                    "As a": parts[0].strip().capitalize(),
+                    "I want to": parts[1].strip().capitalize(),
+                    "So that": parts[1].strip().capitalize()
+                }
+                user_stories.append(story)
     
-    return '. '.join(user_stories)
+    # Format user stories
+    formatted_stories = ""
+    for story in user_stories:
+        formatted_stories += f"**User Story:**\n\n" \
+                             f"**As a:** {story['As a']}\n" \
+                             f"**I want to:** {story['I want to']}\n" \
+                             f"**So that:** {story['So that']}\n\n"
+    
+    return formatted_stories
 
 # Initialize session state for input history if it doesn't exist
 if 'input_history' not in st.session_state: