Update app.py

app.py

@@ -14,7 +14,16 @@ def crack_password(password_hash, wordlist_file, algorithm, salt, parallel, comp
         Cracker.crack_passwords_with_brute_force(min_length, max_length, character_set)
     else:
         Cracker.crack_passwords_with_wordlist()
-    return Cracker.get_statistics()
+    
+    # Получаем статистику
+    stats = Cracker.get_statistics()
+    
+    # Преобразуем статистику в строку для отображения в Streamlit
+    stats_str = f"Total Attempts: {stats.get('total_attempts', 'N/A')}\n"
+    stats_str += f"Cracked Password: {stats.get('cracked_password', 'N/A')}\n"
+    stats_str += f"Time Taken: {stats.get('time_taken', 'N/A')}\n"
+    
+    return stats_str
 
 def main():
     st.title("GVA Password Cracker")
@@ -32,6 +41,10 @@ def main():
     brute_force = st.sidebar.checkbox("Perform Brute Force Attack")
 
     if st.sidebar.button("Crack Password"):
+        if not wordlist_file:
+            st.error("Please upload a wordlist file.")
+            return
+        
         cracking_spinner = st.spinner("Cracking...")
         with cracking_spinner:
             stats = crack_password(
@@ -46,8 +59,8 @@ def main():
                 character_set=character_set,
                 brute_force=brute_force
             )
-        st.success(f"Password Cracked! {stats}")
+        st.success(f"Password Cracked! Statistics data:\n{stats}")
         st.balloons()
 
 if __name__ == "__main__":
-    main()
+    main()