app development tryout

app.py

@@ -11,15 +11,15 @@ model3_generator = pipeline('text2text-generation', model='google/flan-t5-large'
 
 def model1_translate_to_graph(request_text):
     """
-    Model 1: Translate the customer request into a structured graph description.
-    The output should be a JSON-like string with two keys: 'nodes' and 'edges'.
-    Nodes is a list of node names and edges is a list of pairs representing directed edges.
+    Stage 1: Translate the customer request into a structured graph.
+    Output valid JSON with two keys: 'nodes' and 'edges'. 
+    Nodes is a list of unique node names, and edges is a list of pairs [source, target].
     """
     prompt = (
         "Translate the following customer request into a structured graph. "
-        "Output only valid JSON with exactly two keys: 'nodes' (a list of node names) "
-        "and 'edges' (a list of pairs of node names for directed edges). "
-        "Do not include any extra text.\n"
+        "Output only valid JSON with exactly two keys: 'nodes' and 'edges'. "
+        "'nodes' should be a list of unique node names, and 'edges' should be a list of pairs "
+        "of node names representing directed edges. Do not include any extra text.\n"
         f"Customer Request: \"{request_text}\"\n"
         "Structured Graph JSON:"
     )
@@ -28,7 +28,7 @@ def model1_translate_to_graph(request_text):
 
 def model2_generate_response(graph_description):
     """
-    Model 2: Generate a detailed response based on the structured graph description.
+    Stage 2: Generate a detailed response based on the structured graph description.
     """
     prompt = (
         "Based on the following structured customer request graph (formatted as JSON), "
@@ -41,14 +41,13 @@ def model2_generate_response(graph_description):
 
 def model3_response_to_graph(response_text):
     """
-    Model 3: Convert the detailed response into a structured graph description.
+    Stage 3: Convert the detailed response into a structured graph.
     Output only valid JSON with two keys: 'nodes' and 'edges'.
     """
     prompt = (
         "Translate the following response into a structured graph. "
-        "Output only valid JSON with two keys: 'nodes' (a list of concept names) "
-        "and 'edges' (a list of pairs of node names representing relationships). "
-        "Do not include any extra text.\n"
+        "Output only valid JSON with two keys: 'nodes' (a list of concept names) and 'edges' "
+        "(a list of pairs representing relationships between these concepts). Do not include any extra text.\n"
         f"Response: \"{response_text}\"\n"
         "Structured Response Graph JSON:"
     )
@@ -70,8 +69,8 @@ def try_parse_json(text):
 
 def build_and_visualize_graph(graph_data, title="Graph Visualization"):
     """
-    Build a NetworkX graph from a dictionary with 'nodes' and 'edges' keys,
-    and then visualize it using Matplotlib.
+    Build a NetworkX graph from a dictionary with 'nodes' and 'edges'
+    and visualize it using Matplotlib.
     """
     if not graph_data or 'nodes' not in graph_data or 'edges' not in graph_data:
         st.error("Invalid graph data format.")
@@ -89,11 +88,12 @@ def build_and_visualize_graph(graph_data, title="Graph Visualization"):
     st.pyplot(plt)
 
 def main():
-    st.title("Three-Model Pipeline with Flan-T5 and Graph Visualization")
+    st.title("Integrated Three-Stage Pipeline with Flan-T5 and Graph Visualization")
     st.write(
-        "Enter a customer request and see it transformed into a structured graph (using Model 1), "
-        "then used to generate a detailed response (Model 2), and finally converted into a response graph (Model 3) "
-        "that is visualized using NetworkX and Matplotlib."
+        "Enter a customer request to see it processed through three stages:\n\n"
+        "1. **Stage 1:** The request is converted into a structured graph (input graph).\n"
+        "2. **Stage 2:** The graph is used to generate a detailed response.\n"
+        "3. **Stage 3:** The response is converted into another structured graph (output graph) and visualized."
     )
     
     customer_request = st.text_area("Enter Customer Request:", height=150)
@@ -103,25 +103,28 @@ def main():
             st.error("Please enter a valid customer request.")
             return
         
-        st.subheader("Step 1: Customer Request → Input Graph")
+        # --- Stage 1: Request to Graph ---
+        st.subheader("Stage 1: Customer Request → Input Graph")
         model1_output = model1_translate_to_graph(customer_request)
         st.code(model1_output, language="json")
-        graph_input = try_parse_json(model1_output)
-        if graph_input:
-            build_and_visualize_graph(graph_input, title="Input Graph (Customer Request)")
+        input_graph_data = try_parse_json(model1_output)
+        if input_graph_data:
+            build_and_visualize_graph(input_graph_data, title="Input Graph (Customer Request)")
         else:
             st.warning("Could not parse a valid graph structure from Model 1 output.")
         
-        st.subheader("Step 2: Input Graph → Detailed Response")
+        # --- Stage 2: Graph → Detailed Response ---
+        st.subheader("Stage 2: Input Graph → Detailed Response")
         response_text = model2_generate_response(model1_output)
         st.write(response_text)
         
-        st.subheader("Step 3: Detailed Response → Output Graph")
+        # --- Stage 3: Detailed Response → Output Graph ---
+        st.subheader("Stage 3: Detailed Response → Output Graph")
         model3_output = model3_response_to_graph(response_text)
         st.code(model3_output, language="json")
-        graph_output = try_parse_json(model3_output)
-        if graph_output:
-            build_and_visualize_graph(graph_output, title="Output Graph (Structured Response)")
+        output_graph_data = try_parse_json(model3_output)
+        if output_graph_data:
+            build_and_visualize_graph(output_graph_data, title="Output Graph (Structured Response)")
         else:
             st.warning("Could not parse a valid graph structure from Model 3 output.")