Update app.py

app.py

@@ -22,14 +22,17 @@ def compare_structures(struct1, struct2):
 def display_diff(diff):
     left_lines = []
     right_lines = []
+    diff_found = False
     
     for line in diff:
         if line.startswith('- '):
             left_lines.append(f'<span style="background-color: #ffdddd;">{line[2:]}</span>')
             right_lines.append('')
+            diff_found = True
         elif line.startswith('+ '):
             right_lines.append(f'<span style="background-color: #ddffdd;">{line[2:]}</span>')
             left_lines.append('')
+            diff_found = True
         elif line.startswith('  '):
             left_lines.append(line[2:])
             right_lines.append(line[2:])
@@ -39,7 +42,7 @@ def display_diff(diff):
     left_html = "<br>".join(left_lines)
     right_html = "<br>".join(right_lines)
     
-    return left_html, right_html
+    return left_html, right_html, diff_found
 
 st.title("Model Structure Comparison Tool")
 model_id1 = st.text_input("Enter the first HuggingFace Model ID")
@@ -50,18 +53,21 @@ if model_id1 and model_id2:
     struct2 = get_model_structure(model_id2)
     
     diff = compare_structures(struct1, struct2)
-    left_html, right_html = display_diff(diff)
+    left_html, right_html, diff_found = display_diff(diff)
     
     st.write("### Comparison Result")
-    col1, col2 = st.columns([1, 1])  # Adjust the ratio to make columns wider
-    
-    with col1:
-        st.write("### Model 1")
-        st.markdown(left_html, unsafe_allow_html=True)
-    
-    with col2:
-        st.write("### Model 2")
-        st.markdown(right_html, unsafe_allow_html=True)
+    if not diff_found:
+        st.success("The model structures are identical.")
+    else:
+        col1, col2 = st.columns([1.5, 1.5])  # Adjust the ratio to make columns wider
+
+        with col1:
+            st.write("### Model 1")
+            st.markdown(left_html, unsafe_allow_html=True)
+
+        with col2:
+            st.write("### Model 2")
+            st.markdown(right_html, unsafe_allow_html=True)
 
 # Apply custom CSS for wider layout
 st.markdown(