Update app.py

app.py

@@ -11,6 +11,9 @@ import requests
 from fpdf import FPDF
 import markdown
 import tempfile
+import re
+import spacy
+from collections import defaultdict
 
 
 # Database setup
@@ -205,8 +208,6 @@ def workspace_page():
                 except Exception as e:
                     st.error(f"Failed to clone repository: {e}")
 
-#------------------------------------------------------------------------------------------------------------------------------------------------------------------------
-
 
 # Configure Gemini API
 gemini = os.getenv("GEMINI")
@@ -468,6 +469,26 @@ def generate_documentation_page():
                     # Display the final documentation
                     st.success("Documentation generated successfully!")
                     st.text_area("Generated Documentation", documentation, height=600)
+
+                    #---------------------------------------------------------------------------
+                    # Example Gemini output for testing
+                    test_gemini_output = """
+                    Functionality Summary:
+                    Authenticate the user by verifying their credentials. Once authenticated, fetch user-specific data such as preferences and usage history. Handle errors during both authentication and data retrieval.
+
+                    Project Summary:
+                    <Some project details>
+
+                    Functionality Flow:
+                    <Flow details>
+                    """
+
+                    # Streamlit testing
+                    if "page" not in st.session_state or st.session_state.page == "test_functionality":
+                        st.title("Test Functionality Extraction")
+                        display_functionality_analysis(test_gemini_output)
+                    #---------------------------------------------------------------------------
+                    
                 except Exception as e:
                     st.error(f"An error occurred: {e}")
             else:
@@ -475,6 +496,8 @@ def generate_documentation_page():
         else:
             st.error("Please enter the functionality to analyze.")
 
+        
+
     # Add export/download buttons if documentation is available
     if "generated_documentation" in st.session_state and st.session_state.generated_documentation:
         documentation = st.session_state.generated_documentation
@@ -587,6 +610,73 @@ def generate_markdown_file(documentation, output_path):
     with open(output_path, "w") as f:
         f.write("\n".join(formatted_lines))
 
+#---------------------------------------------------------------------------------------------------------------------------------------------------------------------------
+
+# Load the NLP model (use the small model for speed)
+nlp = spacy.load("en_core_web_sm")
+
+def extract_functionality_and_relationships(gemini_output):
+    """
+    Extracts functionality descriptions and sub-functionality relationships from Gemini output.
+    Args:
+        gemini_output (str): The raw output from Gemini.
+    Returns:
+        dict: A dictionary with functionality and its relationships.
+    """
+    functionality_summary = None
+    relationships = defaultdict(list)
+
+    # Extract the 'Functionality Summary' section
+    functionality_match = re.search(r"Functionality Summary:\n(.*?)\n\n", gemini_output, re.DOTALL)
+    if functionality_match:
+        functionality_summary = functionality_match.group(1).strip()
+
+    # Parse the functionality summary with NLP
+    if functionality_summary:
+        doc = nlp(functionality_summary)
+
+        # Extract relationships between sub-functionalities
+        for sentence in doc.sents:
+            sub_functionalities = []
+            for token in sentence:
+                # Capture sub-functionalities using noun chunks or proper nouns
+                if token.dep_ in {"nsubj", "dobj", "pobj"} or token.ent_type_ in {"ORG", "PRODUCT"}:
+                    sub_functionalities.append(token.text)
+            if len(sub_functionalities) > 1:
+                # Create relationships between identified sub-functionalities
+                main_functionality = sub_functionalities[0]
+                for related in sub_functionalities[1:]:
+                    relationships[main_functionality].append(related)
+
+    return functionality_summary, dict(relationships)
+
+
+def display_functionality_analysis(gemini_output):
+    """
+    Extracts and displays functionality analysis for testing purposes.
+    Args:
+        gemini_output (str): The raw output from Gemini.
+    """
+    # Extract functionality and relationships
+    functionality, relationships = extract_functionality_and_relationships(gemini_output)
+
+    # Display extracted functionality
+    st.subheader("Extracted Functionality Summary")
+    if functionality:
+        st.text_area("Functionality Summary", functionality, height=200)
+    else:
+        st.error("No functionality summary found in the Gemini output.")
+
+    # Display extracted relationships
+    st.subheader("Extracted Relationships")
+    if relationships:
+        for main, related in relationships.items():
+            st.write(f"**{main}** is related to: {', '.join(related)}")
+    else:
+        st.info("No relationships between sub-functionalities identified.")
+
+
+
 
 
 #------------------------------------------------------------------------------------------------------------------------------------------------------------------------