bug fix

app.py

@@ -967,23 +967,26 @@ def process_input(
             details = analysis.get('details', "")
             
             img_cyclic = annotate_cyclic_structure(mol, three_letter)
-
-            summary = ""
-            if show_segment_details and details:
-                summary += "Segment Analysis:\n"
-                summary += details + "\n\n"
+            
             summary = "Summary:\n"
             summary += f"Sequence: {three_letter}\n"
             summary += f"One-letter code: {one_letter}\n"
             summary += f"Is Cyclic: {'Yes' if is_cyclic else 'No'}\n"
             
+            # Add segment details if requested
+            if show_segment_details and details:
+                summary += "\n" + "="*50 + "\n"
+                summary += "SEGMENT ANALYSIS:\n"
+                summary += "="*50 + "\n"
+                summary += details + "\n"
+            
             if structure_files:
                 summary += "\n3D Structures Generated:\n"
                 for filepath in structure_files:
                     summary += f"- {os.path.basename(filepath)}\n"
             
             #return summary, img_cyclic, img_linear, structure_files if structure_files else None
-            return summary, img_cyclic, 
+            return summary, img_cyclic, uaa_info
 
         except Exception as e:
             #return f"Error processing SMILES: {str(e)}", None, None, []