Update app.py

app.py

@@ -99,32 +99,43 @@ def generate_html(llm_response):
     #COMPLIANCE ANALYSIS: The proposal complies with the MANUAL requirements as it is based on a firm offer from an OEM, which is acceptable for the specified Tender Type. Since the proposal is for OEM procurement, it adheres to the guidelines that allow for a single vendor's firm offer to serve as the basis for the estimate. There are no deviations noted, and thus no corrective actions are necessary."""
     
     # Parse key-values and analysis
-    data = {'Status': '', 'Severity': '', 'Fix': '', 'Deviations': '', 'Analysis': ''}
-    lines = llm_response.strip().splitlines()
+    # Improved parsing
+    data = {'Status': 'N/A', 'Severity': 'N/A', 'Deviations': 'N/A', 'Fix': 'N/A', 'Analysis': ''}
+    lines = llm_response.strip().split('\n')
+    in_analysis = False
+    analysis_parts = []
+    
     for line in lines:
         line = line.strip()
         if not line:
             continue
-        if ':' in line:
-            key, value = line.split(':', 1)
-            key = key.strip()
-            value = value.strip()
-            if key in data:
-                data[key] = value
+        
+        if line.startswith('Status:'):
+            data['Status'] = line.split(':', 1)[1].strip()
+        elif line.startswith('Severity:'):
+            data['Severity'] = line.split(':', 1)[1].strip()
+        elif line.startswith('Deviations:'):
+            data['Deviations'] = line.split(':', 1)[1].strip()
+        elif line.startswith('Fix:'):
+            data['Fix'] = line.split(':', 1)[1].strip()
         elif 'COMPLIANCE ANALYSIS:' in line:
+            in_analysis = True
             data['Analysis'] = line.split('COMPLIANCE ANALYSIS:', 1)[1].strip()
-        else:
-            data['Analysis'] += ' ' + line  # Append to analysis if multi-line
+        elif in_analysis:
+            analysis_parts.append(line)
+    
+    # Append any trailing analysis
+    if analysis_parts:
+        data['Analysis'] += ' ' + ' '.join(analysis_parts).strip()
     
     #print(data)  # {'Status': 'COMPLIANT', 'Severity': 'LOW', ...}
-    html = """
-                <tr><th>Status</th><td class="status-compliant">{Status}</td></tr>
-                <tr><th>Severity</th><td class="status-low">{Severity}</td></tr>
-                <tr><th>Deviations</th><td>{Deviations}</td></tr>
-                <tr><th>Fix</th><td>{Fix}</td></tr>
-                <tr><th>Analysis</th><td class="analysis">{Analysis}</td></tr>
-            
-        """.format(**data)
+    html = f"""
+        <tr><th>Status</th><td class="compliant">{data.get('Status', 'N/A')}</td></tr>
+        <tr><th>Severity</th><td class="low">{data.get('Severity', 'N/A')}</td></tr>
+        <tr><th>Deviations</th><td>{data.get('Deviations', 'N/A')}</td></tr>
+        <tr><th>Fix</th><td>{data.get('Fix', 'N/A')}</td></tr>
+        <tr><th>Analysis</th><td class="analysis">{data.get('Analysis', 'N/A')}</td></tr>            
+        """
     return html